Implement Random Forest nodes/leaves/depth

2023-11-28 00:35:38 +01:00
parent 4addaefb47
commit d06bf187b2
6 changed files with 60 additions and 1 deletions
--- a/src/PyClassifiers/RandomForest.h
+++ b/src/PyClassifiers/RandomForest.h
@@ -7,6 +7,9 @@ namespace pywrap {
    public:
        RandomForest();
        ~RandomForest() = default;
+        int getNumberOfEdges() const override;
+        int getNumberOfStates() const override;
+        int getNumberOfNodes() const override;
    };
 } /* namespace pywrap */
 #endif /* RANDOMFOREST_H */