From 2163e95c4afb5a4a5943cc5e71ead9061f6d39c0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Fri, 9 Feb 2024 10:57:19 +0100
Subject: [PATCH] add getNotes method

---
 src/BayesNet/BaseClassifier.h | 1 +
 src/BayesNet/Classifier.h     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/BayesNet/BaseClassifier.h b/src/BayesNet/BaseClassifier.h
index f46054d..e84f494 100644
--- a/src/BayesNet/BaseClassifier.h
+++ b/src/BayesNet/BaseClassifier.h
@@ -26,6 +26,7 @@ namespace bayesnet {
         std::vector<std::string> virtual graph(const std::string& title = "") const = 0;
         virtual std::string getVersion() = 0;
         std::vector<std::string> virtual topological_order() = 0;
+        std::vector<std::string> virtual getNotes() const = 0;
         void virtual dump_cpt()const = 0;
         virtual void setHyperparameters(const nlohmann::json& hyperparameters) = 0;
         std::vector<std::string>& getValidHyperparameters() { return validHyperparameters; }
diff --git a/src/BayesNet/Classifier.h b/src/BayesNet/Classifier.h
index 9db25d9..3dfd18b 100644
--- a/src/BayesNet/Classifier.h
+++ b/src/BayesNet/Classifier.h
@@ -43,6 +43,7 @@ namespace bayesnet {
         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 { return notes; }
         void dump_cpt() const override;
         void setHyperparameters(const nlohmann::json& hyperparameters) override; //For classifiers that don't have hyperparameters
     };