Add getNotes method

2024-02-09 11:26:39 +01:00
parent aa2b09063e
commit cb548ff221
1 changed files with 2 additions and 0 deletions
--- 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<std::string> topological_order() override { return std::vector<std::string>(); }
        void dump_cpt() const override {};
+        std::vector<std::string> 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<std::string> notes;
    private:
        PyWrap* pyWrap;
        std::string module;