![]() |
BayesNet 1.0.5
Bayesian Network Classifiers using libtorch from scratch
|
Public Member Functions | |
Boost (bool predict_voting=false) | |
void | setHyperparameters (const nlohmann::json &hyperparameters_) override |
![]() | |
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 |
![]() | |
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) 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) override |
Classifier & | fit (torch::Tensor &dataset, const std::vector< std::string > &features, const std::string &className, std::map< std::string, std::vector< int > > &states) 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) 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 |
![]() | |
std::vector< std::string > & | getValidHyperparameters () |
Protected Member Functions | |
std::vector< int > | featureSelection (torch::Tensor &weights_) |
void | buildModel (const torch::Tensor &weights) override |
std::tuple< torch::Tensor &, double, bool > | update_weights (torch::Tensor &ytrain, torch::Tensor &ypred, torch::Tensor &weights) |
std::tuple< torch::Tensor &, double, bool > | update_weights_block (int k, torch::Tensor &ytrain, torch::Tensor &weights) |
![]() | |
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) |
void | trainModel (const torch::Tensor &weights) override |
![]() | |
void | checkFitParameters () |
void | buildDataset (torch::Tensor &y) |
Protected Attributes | |
torch::Tensor | X_train |
torch::Tensor | y_train |
torch::Tensor | X_test |
torch::Tensor | y_test |
bool | bisection = true |
int | maxTolerance = 3 |
std::string | order_algorithm |
bool | convergence = true |
bool | convergence_best = false |
bool | selectFeatures = false |
std::string | select_features_algorithm = Orders.DESC |
FeatureSelect * | featureSelector = nullptr |
double | threshold = -1 |
bool | block_update = false |
![]() | |
unsigned | n_models |
std::vector< std::unique_ptr< Classifier > > | models |
std::vector< double > | significanceModels |
bool | predict_voting |
![]() | |
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 |
status_t | status = NORMAL |
std::vector< std::string > | notes |
![]() | |
std::vector< std::string > | validHyperparameters |
|
explicit |
|
overrideprotectedvirtual |
Implements bayesnet::Classifier.
|
protected |
|
overridevirtual |
Implements bayesnet::BaseClassifier.
|
protected |
|
protected |
|
protected |
|
protected |