Refactor version method in PyClassifier

2023-11-13 13:59:06 +01:00
parent 431b3a3aa5
commit 69ad660040
7 changed files with 14 additions and 18 deletions
--- a/src/PyClassifiers/RandomForest.h
+++ b/src/PyClassifiers/RandomForest.h
@@ -5,9 +5,9 @@
 namespace pywrap {
    class RandomForest : public PyClassifier {
    public:
-        RandomForest() : PyClassifier("sklearn.ensemble", "RandomForestClassifier") {};
+        RandomForest() : PyClassifier("sklearn.ensemble", "RandomForestClassifier", true) {};
        ~RandomForest() = default;
-        std::string version();
+        void setHyperparameters(nlohmann::json& hyperparameters) override;
    };
 } /* namespace pywrap */
 #endif /* RANDOMFOREST_H */