bayesnet::Classifier Class Reference

Inheritance diagram for bayesnet::Classifier:

Collaboration diagram for bayesnet::Classifier:

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, const Smoothing_t smoothing) 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, const Smoothing_t smoothing) 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 Smoothing_t smoothing) 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, const Smoothing_t smoothing) 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, const Smoothing_t smoothing) override
void	buildDataset (torch::Tensor &y)
Protected Member Functions inherited from bayesnet::BaseClassifier

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
const std::string	CLASSIFIER_NOT_FITTED = "Classifier has not been fitted"
Protected Attributes inherited from bayesnet::BaseClassifier
std::vector< std::string >	validHyperparameters
std::vector< std::string >	notes
status_t	status = NORMAL

Detailed Description

Definition at line 14 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 154 of file Classifier.cc.

buildDataset()

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

protected

Definition at line 29 of file Classifier.cc.

checkFitParameters()

void bayesnet::Classifier::checkFitParameters ( )

protected

Definition at line 78 of file Classifier.cc.

dump_cpt()

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 183 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, const Smoothing_t smoothing )

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 56 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, const Smoothing_t smoothing )

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 67 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, const Smoothing_t smoothing )

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 73 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, const Smoothing_t smoothing )

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 48 of file Classifier.cc.

getClassNumStates()

int bayesnet::Classifier::getClassNumStates ( ) const

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 175 of file Classifier.cc.

getNotes()

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

inlineoverridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 37 of file Classifier.h.

getNumberOfEdges()

int bayesnet::Classifier::getNumberOfEdges ( ) const

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 167 of file Classifier.cc.

getNumberOfNodes()

int bayesnet::Classifier::getNumberOfNodes ( ) const

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 162 of file Classifier.cc.

getNumberOfStates()

int bayesnet::Classifier::getNumberOfStates ( ) const

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 171 of file Classifier.cc.

getStatus()

status_t bayesnet::Classifier::getStatus ( ) const

inlineoverridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 31 of file Classifier.h.

getVersion()

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

inlineoverridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 32 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 102 of file Classifier.cc.

predict() [2/2]

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 95 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 123 of file Classifier.cc.

predict_proba() [2/2]

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 116 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 143 of file Classifier.cc.

score() [2/2]

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 138 of file Classifier.cc.

setHyperparameters()

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 187 of file Classifier.cc.

show()

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 150 of file Classifier.cc.

topological_order()

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 179 of file Classifier.cc.

trainModel()

void bayesnet::Classifier::trainModel ( const torch::Tensor & weights, const Smoothing_t smoothing )

overrideprotectedvirtual

Implements bayesnet::BaseClassifier.

Definition at line 43 of file Classifier.cc.

Member Data Documentation

CLASSIFIER_NOT_FITTED

const std::string bayesnet::Classifier::CLASSIFIER_NOT_FITTED = "Classifier has not been fitted"

protected

Definition at line 53 of file Classifier.h.

className

std::string bayesnet::Classifier::className

protected

Definition at line 46 of file Classifier.h.

dataset

torch::Tensor bayesnet::Classifier::dataset

protected

Definition at line 48 of file Classifier.h.

features

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

protected

Definition at line 45 of file Classifier.h.

fitted

bool bayesnet::Classifier::fitted

protected

Definition at line 41 of file Classifier.h.

m

unsigned int bayesnet::Classifier::m

protected

Definition at line 42 of file Classifier.h.

metrics

Metrics bayesnet::Classifier::metrics

protected

Definition at line 44 of file Classifier.h.

model

Network bayesnet::Classifier::model

protected

Definition at line 43 of file Classifier.h.

n

unsigned int bayesnet::Classifier::n

protected

Definition at line 42 of file Classifier.h.

states

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

protected

Definition at line 47 of file Classifier.h.

---

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

• /home/rmontanana/Code/BayesNet/bayesnet/classifiers/Classifier.h
• /home/rmontanana/Code/BayesNet/bayesnet/classifiers/Classifier.cc