Add nodes/leaves/depth to STree & ODTE

2023-11-27 10:57:57 +01:00
parent 4fefe9a1d2
commit 82964190f6
8 changed files with 53 additions and 2 deletions
--- a/src/PyClassifiers/ODTE.h
+++ b/src/PyClassifiers/ODTE.h
@@ -8,6 +8,9 @@ namespace pywrap {
    public:
        ODTE();
        ~ODTE() = default;
+        int getNumberOfNodes() const override;
+        int getNumberOfEdges() const override;
+        int getNumberOfStates() const override;
        std::string graph();
    };
 } /* namespace pywrap */