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

Public Member Functions

TANLd & fit (torch::Tensor &X, torch::Tensor &y, const std::vector< std::string > &features, const std::string &className, map< std::string, std::vector< int > > &states, const Smoothing_t smoothing) override
 
std::vector< std::string > graph (const std::string &name="TANLd") const override
 
torch::Tensor predict (torch::Tensor &X) override
 
- Public Member Functions inherited from bayesnet::TAN
void setHyperparameters (const nlohmann::json &hyperparameters_) 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
 
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
 
- Public Member Functions inherited from bayesnet::BaseClassifier
std::vector< std::string > & getValidHyperparameters ()
 
- Public Member Functions inherited from bayesnet::Proposal
 Proposal (torch::Tensor &pDataset, std::vector< std::string > &features_, std::string &className_)
 

Additional Inherited Members

- Protected Member Functions inherited from bayesnet::TAN
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
- Protected Member Functions inherited from bayesnet::Proposal
void checkInput (const torch::Tensor &X, const torch::Tensor &y)
 
torch::Tensor prepareX (torch::Tensor &X)
 
map< std::string, std::vector< int > > localDiscretizationProposal (const map< std::string, std::vector< int > > &states, Network &model)
 
map< std::string, std::vector< int > > fit_local_discretization (const torch::Tensor &y)
 
- 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
 
- Protected Attributes inherited from bayesnet::Proposal
torch::Tensor Xf
 
torch::Tensor y
 
map< std::string, mdlp::CPPFImdlp * > discretizers
 

Detailed Description

Definition at line 13 of file TANLd.h.

Constructor & Destructor Documentation

◆ TANLd()

bayesnet::TANLd::TANLd ( )

Definition at line 10 of file TANLd.cc.

Member Function Documentation

◆ fit()

TANLd & bayesnet::TANLd::fit ( torch::Tensor & X,
torch::Tensor & y,
const std::vector< std::string > & features,
const std::string & className,
map< std::string, std::vector< int > > & states,
const Smoothing_t smoothing )
override

Definition at line 11 of file TANLd.cc.

◆ graph()

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

Reimplemented from bayesnet::TAN.

Definition at line 32 of file TANLd.cc.

◆ predict()

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

Reimplemented from bayesnet::Classifier.

Definition at line 27 of file TANLd.cc.

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