Refactor library structure

2024-03-08 22:20:38 +01:00
parent 5bec9d4d2f
commit 7534cba7e6
23 changed files with 22 additions and 152 deletions
--- a/pyclfs/ODTE.cc
+++ b/pyclfs/ODTE.cc
@@ -0,0 +1,24 @@
+#include "ODTE.h"
+
+namespace pywrap {
+    ODTE::ODTE() : PyClassifier("odte", "Odte")
+    {
+        validHyperparameters = { "n_jobs", "n_estimators", "random_state" };
+    }
+    int ODTE::getNumberOfNodes() const
+    {
+        return callMethodInt("get_nodes");
+    }
+    int ODTE::getNumberOfEdges() const
+    {
+        return callMethodInt("get_leaves");
+    }
+    int ODTE::getNumberOfStates() const
+    {
+        return callMethodInt("get_depth");
+    }
+    std::string ODTE::graph()
+    {
+        return callMethodString("graph");
+    }
+} /* namespace pywrap */