rmontanana/Pyclassifiers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add getNotes method

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-02-09 11:26:39 +01:00
parent aa2b09063e
commit cb548ff221
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/PyClassifiers/PyClassifier.h
View File

@@ -40,10 +40,12 @@ namespace pywrap {
bayesnet::status_t getStatus() const override { return bayesnet::NORMAL; }; bayesnet::status_t getStatus() const override { return bayesnet::NORMAL; };
std::vector<std::string> topological_order() override { return std::vector<std::string>(); } std::vector<std::string> topological_order() override { return std::vector<std::string>(); }
void dump_cpt() const override {}; void dump_cpt() const override {};
std::vector<std::string> getNotes() const override { return notes; };
void setHyperparameters(const nlohmann::json& hyperparameters) override; void setHyperparameters(const nlohmann::json& hyperparameters) override;
protected: protected:
nlohmann::json hyperparameters; nlohmann::json hyperparameters;
void trainModel(const torch::Tensor& weights) override {}; void trainModel(const torch::Tensor& weights) override {};
std::vector<std::string> notes;
private: private:
PyWrap* pyWrap; PyWrap* pyWrap;
std::string module; std::string module;