Fix classifier build in proposal

2025-07-07 02:10:08 +02:00
parent 0ce7f664b4
commit 2c7352ac38
7 changed files with 29 additions and 16 deletions
--- a/bayesnet/classifiers/Classifier.h
+++ b/bayesnet/classifiers/Classifier.h
@@ -37,6 +37,7 @@ namespace bayesnet {
        std::vector<std::string> getNotes() const override { return notes; }
        std::string dump_cpt() const override;
        void setHyperparameters(const nlohmann::json& hyperparameters) override; //For classifiers that don't have hyperparameters
+        Network& getModel() { return model; }
    protected:
        bool fitted;
        unsigned int m, n; // m: number of samples, n: number of features