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/PyWrap.h
+++ b/src/PyClassifiers/PyWrap.h
@@ -24,6 +24,7 @@ namespace pywrap {
        void operator=(const PyWrap&) = delete;
        ~PyWrap() = default;
        std::string callMethodString(const clfId_t id, const std::string& method);
+        int callMethodInt(const clfId_t id, const std::string& method);
        std::string sklearnVersion();
        std::string version(const clfId_t id);
        void setHyperparameters(const clfId_t id, const json& hyperparameters);