|
| KDB (int k, float theta=0.03) |
|
void | setHyperparameters (const nlohmann::json &hyperparameters_) override |
|
std::vector< std::string > | graph (const std::string &name="KDB") 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, 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 |
|
std::vector< std::string > & | getValidHyperparameters () |
|
Definition at line 12 of file KDB.h.