Inheritance diagram for bayesnet::AODE:

[legend]

Collaboration diagram for bayesnet::AODE:

[legend]

Public Member Functions
	AODE (bool predict_voting=false)

void	setHyperparameters (const nlohmann::json &hyperparameters) override

std::vector< std::string >	graph (const std::string &title="AODE") const override

Public Member Functions inherited from bayesnet::Ensemble
	Ensemble (bool predict_voting=true)

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

float	score (torch::Tensor &X, torch::Tensor &y) override

float	score (std::vector< std::vector< int > > &X, std::vector< int > &y) override

int	getNumberOfNodes () const override

int	getNumberOfEdges () const override

int	getNumberOfStates () const override

std::vector< std::string >	show () const override

std::vector< std::string >	graph (const std::string &title) const override

std::vector< std::string >	topological_order () override

std::string	dump_cpt () const override

Public Member Functions inherited from bayesnet::Classifier
	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	getClassNumStates () const override

status_t	getStatus () const override

std::string	getVersion () override

std::vector< std::string >	getNotes () const override

void	setHyperparameters (const nlohmann::json &hyperparameters) override

Public Member Functions inherited from bayesnet::BaseClassifier
std::vector< std::string > &	getValidHyperparameters ()

Protected Member Functions
void	buildModel (const torch::Tensor &weights) override

Protected Member Functions inherited from bayesnet::Ensemble
void	trainModel (const torch::Tensor &weights, const Smoothing_t smoothing) override

torch::Tensor	predict_average_voting (torch::Tensor &X)

std::vector< std::vector< double > >	predict_average_voting (std::vector< std::vector< int > > &X)

torch::Tensor	predict_average_proba (torch::Tensor &X)

std::vector< std::vector< double > >	predict_average_proba (std::vector< std::vector< int > > &X)

torch::Tensor	compute_arg_max (torch::Tensor &X)

std::vector< int >	compute_arg_max (std::vector< std::vector< double > > &X)

torch::Tensor	voting (torch::Tensor &votes)

Protected Member Functions inherited from bayesnet::Classifier
void	checkFitParameters ()

void	buildDataset (torch::Tensor &y)

Protected Member Functions inherited from bayesnet::BaseClassifier

Additional Inherited Members
Protected Attributes inherited from bayesnet::Ensemble
unsigned	n_models

std::vector< std::unique_ptr< Classifier > >	models

std::vector< double >	significanceModels

bool	predict_voting

Protected Attributes inherited from bayesnet::Classifier
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 12 of file AODE.h.

Constructor & Destructor Documentation

◆ AODE()

bayesnet::AODE::AODE ( bool predict_voting = false )

Definition at line 10 of file AODE.cc.

◆ ~AODE()

virtual bayesnet::AODE::~AODE ( )

inlinevirtual

Definition at line 15 of file AODE.h.

Member Function Documentation

◆ buildModel()

void bayesnet::AODE::buildModel ( const torch::Tensor & weights )

overrideprotectedvirtual

Implements bayesnet::Classifier.

Definition at line 24 of file AODE.cc.

◆ graph()

std::vector< std::string > bayesnet::AODE::graph ( const std::string & title = "AODE" ) const

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 34 of file AODE.cc.

◆ setHyperparameters()

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

overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 15 of file AODE.cc.

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

/home/rmontanana/Code/BayesNet/bayesnet/ensembles/AODE.h
/home/rmontanana/Code/BayesNet/bayesnet/ensembles/AODE.cc

Public Member Functions

Protected Member Functions

Additional Inherited Members