BayesNet 1.0.7.
Bayesian Network and basic classifiers Library.
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
bayesnet::Ensemble Class Reference
Inheritance diagram for bayesnet::Ensemble:
Inheritance graph
[legend]
Collaboration diagram for bayesnet::Ensemble:
Collaboration graph
[legend]

Public Member Functions

 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 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 ()
 
virtual void buildModel (const torch::Tensor &weights)=0
 
void buildDataset (torch::Tensor &y)
 
- Protected Member Functions inherited from bayesnet::BaseClassifier

Protected Attributes

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 15 of file Ensemble.h.

Constructor & Destructor Documentation

◆ Ensemble()

bayesnet::Ensemble::Ensemble ( bool predict_voting = true)

Definition at line 10 of file Ensemble.cc.

Member Function Documentation

◆ compute_arg_max() [1/2]

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

Definition at line 22 of file Ensemble.cc.

◆ compute_arg_max() [2/2]

torch::Tensor bayesnet::Ensemble::compute_arg_max ( torch::Tensor & X)
protected

Definition at line 31 of file Ensemble.cc.

◆ dump_cpt()

std::string bayesnet::Ensemble::dump_cpt ( ) const
inlineoverridevirtual

Reimplemented from bayesnet::Classifier.

Definition at line 34 of file Ensemble.h.

◆ getNumberOfEdges()

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

Reimplemented from bayesnet::Classifier.

Definition at line 181 of file Ensemble.cc.

◆ getNumberOfNodes()

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

Reimplemented from bayesnet::Classifier.

Definition at line 173 of file Ensemble.cc.

◆ getNumberOfStates()

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

Reimplemented from bayesnet::Classifier.

Definition at line 189 of file Ensemble.cc.

◆ graph()

std::vector< std::string > bayesnet::Ensemble::graph ( const std::string & title) const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 164 of file Ensemble.cc.

◆ predict() [1/2]

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

Reimplemented from bayesnet::Classifier.

Definition at line 72 of file Ensemble.cc.

◆ predict() [2/2]

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

Reimplemented from bayesnet::Classifier.

Definition at line 77 of file Ensemble.cc.

◆ predict_average_proba() [1/2]

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

Definition at line 95 of file Ensemble.cc.

◆ predict_average_proba() [2/2]

torch::Tensor bayesnet::Ensemble::predict_average_proba ( torch::Tensor & X)
protected

Definition at line 82 of file Ensemble.cc.

◆ predict_average_voting() [1/2]

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

Definition at line 116 of file Ensemble.cc.

◆ predict_average_voting() [2/2]

torch::Tensor bayesnet::Ensemble::predict_average_voting ( torch::Tensor & X)
protected

Definition at line 123 of file Ensemble.cc.

◆ predict_proba() [1/2]

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

Reimplemented from bayesnet::Classifier.

Definition at line 58 of file Ensemble.cc.

◆ predict_proba() [2/2]

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

Reimplemented from bayesnet::Classifier.

Definition at line 65 of file Ensemble.cc.

◆ score() [1/2]

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

Reimplemented from bayesnet::Classifier.

Definition at line 144 of file Ensemble.cc.

◆ score() [2/2]

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

Reimplemented from bayesnet::Classifier.

Definition at line 133 of file Ensemble.cc.

◆ show()

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

Reimplemented from bayesnet::Classifier.

Definition at line 155 of file Ensemble.cc.

◆ topological_order()

std::vector< std::string > bayesnet::Ensemble::topological_order ( )
inlineoverridevirtual

Reimplemented from bayesnet::Classifier.

Definition at line 30 of file Ensemble.h.

◆ trainModel()

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

Reimplemented from bayesnet::Classifier.

Definition at line 14 of file Ensemble.cc.

◆ voting()

torch::Tensor bayesnet::Ensemble::voting ( torch::Tensor & votes)
protected

Definition at line 36 of file Ensemble.cc.

Member Data Documentation

◆ models

std::vector<std::unique_ptr<Classifier> > bayesnet::Ensemble::models
protected

Definition at line 54 of file Ensemble.h.

◆ n_models

unsigned bayesnet::Ensemble::n_models
protected

Definition at line 53 of file Ensemble.h.

◆ predict_voting

bool bayesnet::Ensemble::predict_voting
protected

Definition at line 56 of file Ensemble.h.

◆ significanceModels

std::vector<double> bayesnet::Ensemble::significanceModels
protected

Definition at line 55 of file Ensemble.h.

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