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::Classifier Class Referenceabstract
Inheritance diagram for bayesnet::Classifier:
Inheritance graph
[legend]
Collaboration diagram for bayesnet::Classifier:
Collaboration graph
[legend]

Public Member Functions

 Classifier (Network model)
 
Classifier & 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) override
 
Classifier & 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) override
 
Classifier & fit (torch::Tensor &dataset, const std::vector< std::string > &features, const std::string &className, std::map< std::string, std::vector< int > > &states) override
 
Classifier & 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) override
 
void addNodes ()
 
int getNumberOfNodes () const override
 
int getNumberOfEdges () const override
 
int getNumberOfStates () const override
 
int getClassNumStates () const override
 
torch::Tensor predict (torch::Tensor &X) override
 
std::vector< int > predict (std::vector< std::vector< int > > &X) override
 
torch::Tensor predict_proba (torch::Tensor &X) override
 
std::vector< std::vector< double > > predict_proba (std::vector< std::vector< int > > &X) override
 
status_t getStatus () const override
 
std::string getVersion () override
 
float score (torch::Tensor &X, torch::Tensor &y) override
 
float score (std::vector< std::vector< int > > &X, std::vector< int > &y) override
 
std::vector< std::string > show () const override
 
std::vector< std::string > topological_order () override
 
std::vector< std::string > getNotes () const override
 
std::string dump_cpt () const override
 
void setHyperparameters (const nlohmann::json &hyperparameters) override
 
- Public Member Functions inherited from bayesnet::BaseClassifier
virtual std::vector< std::string > graph (const std::string &title="") const =0
 
std::vector< std::string > & getValidHyperparameters ()
 

Protected Member Functions

void checkFitParameters ()
 
virtual void buildModel (const torch::Tensor &weights)=0
 
void trainModel (const torch::Tensor &weights) override
 
void buildDataset (torch::Tensor &y)
 

Protected Attributes

bool fitted
 
unsigned int m
 
unsigned int n
 
Network model
 
Metrics metrics
 
std::vector< std::string > features
 
std::string className
 
std::map< std::string, std::vector< int > > states
 
torch::Tensor dataset
 
status_t status = NORMAL
 
std::vector< std::string > notes
 
- Protected Attributes inherited from bayesnet::BaseClassifier
std::vector< std::string > validHyperparameters
 

Detailed Description

Definition at line 15 of file Classifier.h.

Constructor & Destructor Documentation

◆ Classifier()

bayesnet::Classifier::Classifier ( Network model)

Definition at line 12 of file Classifier.cc.

Member Function Documentation

◆ addNodes()

void bayesnet::Classifier::addNodes ( )

Definition at line 155 of file Classifier.cc.

◆ buildDataset()

void bayesnet::Classifier::buildDataset ( torch::Tensor & y)
protected

Definition at line 30 of file Classifier.cc.

◆ checkFitParameters()

void bayesnet::Classifier::checkFitParameters ( )
protected

Definition at line 79 of file Classifier.cc.

◆ dump_cpt()

std::string bayesnet::Classifier::dump_cpt ( ) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 184 of file Classifier.cc.

◆ fit() [1/4]

Classifier & bayesnet::Classifier::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 )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 57 of file Classifier.cc.

◆ fit() [2/4]

Classifier & bayesnet::Classifier::fit ( torch::Tensor & dataset,
const std::vector< std::string > & features,
const std::string & className,
std::map< std::string, std::vector< int > > & states )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 68 of file Classifier.cc.

◆ fit() [3/4]

Classifier & bayesnet::Classifier::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 )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 74 of file Classifier.cc.

◆ fit() [4/4]

Classifier & bayesnet::Classifier::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 )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 49 of file Classifier.cc.

◆ getClassNumStates()

int bayesnet::Classifier::getClassNumStates ( ) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 176 of file Classifier.cc.

◆ getNotes()

std::vector< std::string > bayesnet::Classifier::getNotes ( ) const
inlineoverridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 38 of file Classifier.h.

◆ getNumberOfEdges()

int bayesnet::Classifier::getNumberOfEdges ( ) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 168 of file Classifier.cc.

◆ getNumberOfNodes()

int bayesnet::Classifier::getNumberOfNodes ( ) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 163 of file Classifier.cc.

◆ getNumberOfStates()

int bayesnet::Classifier::getNumberOfStates ( ) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 172 of file Classifier.cc.

◆ getStatus()

status_t bayesnet::Classifier::getStatus ( ) const
inlineoverridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 32 of file Classifier.h.

◆ getVersion()

std::string bayesnet::Classifier::getVersion ( )
inlineoverridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 33 of file Classifier.h.

◆ predict() [1/2]

std::vector< int > bayesnet::Classifier::predict ( std::vector< std::vector< int > > & X)
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 103 of file Classifier.cc.

◆ predict() [2/2]

torch::Tensor bayesnet::Classifier::predict ( torch::Tensor & X)
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 96 of file Classifier.cc.

◆ predict_proba() [1/2]

std::vector< std::vector< double > > bayesnet::Classifier::predict_proba ( std::vector< std::vector< int > > & X)
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 124 of file Classifier.cc.

◆ predict_proba() [2/2]

torch::Tensor bayesnet::Classifier::predict_proba ( torch::Tensor & X)
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 117 of file Classifier.cc.

◆ score() [1/2]

float bayesnet::Classifier::score ( std::vector< std::vector< int > > & X,
std::vector< int > & y )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 144 of file Classifier.cc.

◆ score() [2/2]

float bayesnet::Classifier::score ( torch::Tensor & X,
torch::Tensor & y )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 139 of file Classifier.cc.

◆ setHyperparameters()

void bayesnet::Classifier::setHyperparameters ( const nlohmann::json & hyperparameters)
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 188 of file Classifier.cc.

◆ show()

std::vector< std::string > bayesnet::Classifier::show ( ) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 151 of file Classifier.cc.

◆ topological_order()

std::vector< std::string > bayesnet::Classifier::topological_order ( )
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 180 of file Classifier.cc.

◆ trainModel()

void bayesnet::Classifier::trainModel ( const torch::Tensor & weights)
overrideprotectedvirtual

Implements bayesnet::BaseClassifier.

Definition at line 44 of file Classifier.cc.

Member Data Documentation

◆ className

std::string bayesnet::Classifier::className
protected

Definition at line 47 of file Classifier.h.

◆ dataset

torch::Tensor bayesnet::Classifier::dataset
protected

Definition at line 49 of file Classifier.h.

◆ features

std::vector<std::string> bayesnet::Classifier::features
protected

Definition at line 46 of file Classifier.h.

◆ fitted

bool bayesnet::Classifier::fitted
protected

Definition at line 42 of file Classifier.h.

◆ m

unsigned int bayesnet::Classifier::m
protected

Definition at line 43 of file Classifier.h.

◆ metrics

Metrics bayesnet::Classifier::metrics
protected

Definition at line 45 of file Classifier.h.

◆ model

Network bayesnet::Classifier::model
protected

Definition at line 44 of file Classifier.h.

◆ n

unsigned int bayesnet::Classifier::n
protected

Definition at line 43 of file Classifier.h.

◆ notes

std::vector<std::string> bayesnet::Classifier::notes
protected

Definition at line 51 of file Classifier.h.

◆ states

std::map<std::string, std::vector<int> > bayesnet::Classifier::states
protected

Definition at line 48 of file Classifier.h.

◆ status

status_t bayesnet::Classifier::status = NORMAL
protected

Definition at line 50 of file Classifier.h.

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