BayesNet 1.0.5
Bayesian Network Classifiers using libtorch from scratch
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
bayesnet::BaseClassifier Class Referenceabstract
Inheritance diagram for bayesnet::BaseClassifier:
Inheritance graph
[legend]

Public Member Functions

virtual BaseClassifier & fit (std::vector< std::vector< int > > &X, std::vector< int > &y, const std::vector< std::string > &features, const std::string &className, std::map< std::string, std::vector< int > > &states)=0
 
virtual BaseClassifier & fit (torch::Tensor &X, torch::Tensor &y, const std::vector< std::string > &features, const std::string &className, std::map< std::string, std::vector< int > > &states)=0
 
virtual BaseClassifier & fit (torch::Tensor &dataset, const std::vector< std::string > &features, const std::string &className, std::map< std::string, std::vector< int > > &states)=0
 
virtual BaseClassifier & fit (torch::Tensor &dataset, const std::vector< std::string > &features, const std::string &className, std::map< std::string, std::vector< int > > &states, const torch::Tensor &weights)=0
 
virtual torch::Tensor predict (torch::Tensor &X)=0
 
virtual std::vector< int > predict (std::vector< std::vector< int > > &X)=0
 
virtual torch::Tensor predict_proba (torch::Tensor &X)=0
 
virtual std::vector< std::vector< double > > predict_proba (std::vector< std::vector< int > > &X)=0
 
virtual status_t getStatus () const =0
 
virtual float score (std::vector< std::vector< int > > &X, std::vector< int > &y)=0
 
virtual float score (torch::Tensor &X, torch::Tensor &y)=0
 
virtual int getNumberOfNodes () const =0
 
virtual int getNumberOfEdges () const =0
 
virtual int getNumberOfStates () const =0
 
virtual int getClassNumStates () const =0
 
virtual std::vector< std::string > show () const =0
 
virtual std::vector< std::string > graph (const std::string &title="") const =0
 
virtual std::string getVersion ()=0
 
virtual std::vector< std::string > topological_order ()=0
 
virtual std::vector< std::string > getNotes () const =0
 
virtual std::string dump_cpt () const =0
 
virtual void setHyperparameters (const nlohmann::json &hyperparameters)=0
 
std::vector< std::string > & getValidHyperparameters ()
 

Protected Member Functions

virtual void trainModel (const torch::Tensor &weights)=0
 

Protected Attributes

std::vector< std::string > validHyperparameters
 

Detailed Description

Definition at line 13 of file BaseClassifier.h.

Member Function Documentation

◆ getValidHyperparameters()

std::vector< std::string > & bayesnet::BaseClassifier::getValidHyperparameters ( )
inline

Definition at line 40 of file BaseClassifier.h.

Member Data Documentation

◆ validHyperparameters

std::vector<std::string> bayesnet::BaseClassifier::validHyperparameters
protected

Definition at line 43 of file BaseClassifier.h.

The documentation for this class was generated from the following file: