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

Public Member Functions

 SPnDE (std::vector< int > parents)
 
std::vector< std::string > graph (const std::string &name="SPnDE") 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 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
std::vector< std::string > & getValidHyperparameters ()
 

Protected Member Functions

void buildModel (const torch::Tensor &weights) override
 
- Protected Member Functions inherited from bayesnet::Classifier
void checkFitParameters ()
 
void trainModel (const torch::Tensor &weights, const Smoothing_t smoothing) override
 
void buildDataset (torch::Tensor &y)
 
- Protected Member Functions inherited from bayesnet::BaseClassifier

Additional Inherited Members

- 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 13 of file SPnDE.h.

Constructor & Destructor Documentation

◆ SPnDE()

bayesnet::SPnDE::SPnDE ( std::vector< int > parents)
explicit

Definition at line 11 of file SPnDE.cc.

Member Function Documentation

◆ buildModel()

void bayesnet::SPnDE::buildModel ( const torch::Tensor & weights)
overrideprotectedvirtual

Implements bayesnet::Classifier.

Definition at line 13 of file SPnDE.cc.

◆ graph()

std::vector< std::string > bayesnet::SPnDE::graph ( const std::string & name = "SPnDE") const
overridevirtual

Implements bayesnet::BaseClassifier.

Definition at line 33 of file SPnDE.cc.

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