diff --git a/src/PyClassifiers/PyClassifier.h b/src/PyClassifiers/PyClassifier.h index 7260d2e..0686546 100644 --- a/src/PyClassifiers/PyClassifier.h +++ b/src/PyClassifiers/PyClassifier.h @@ -40,10 +40,12 @@ namespace pywrap { bayesnet::status_t getStatus() const override { return bayesnet::NORMAL; }; std::vector topological_order() override { return std::vector(); } void dump_cpt() const override {}; + std::vector getNotes() const override { return notes; }; void setHyperparameters(const nlohmann::json& hyperparameters) override; protected: nlohmann::json hyperparameters; void trainModel(const torch::Tensor& weights) override {}; + std::vector notes; private: PyWrap* pyWrap; std::string module;