update to BayesNet 1.0.3

2024-02-25 18:02:36 +01:00
parent 5a6d38e900
commit bd6f6f5837
21 changed files with 37 additions and 48 deletions
--- a/src/STree.cc
+++ b/src/STree.cc
@@ -0,0 +1,24 @@
+#include "STree.h"
+
+namespace pywrap {
+    STree::STree() : PyClassifier("stree", "Stree")
+    {
+        validHyperparameters = { "C", "kernel", "max_iter", "max_depth", "random_state", "multiclass_strategy", "gamma", "max_features", "degree" };
+    };
+    int STree::getNumberOfNodes() const
+    {
+        return callMethodInt("get_nodes");
+    }
+    int STree::getNumberOfEdges() const
+    {
+        return callMethodInt("get_leaves");
+    }
+    int STree::getNumberOfStates() const
+    {
+        return callMethodInt("get_depth");
+    }
+    std::string STree::graph()
+    {
+        return callMethodString("graph");
+    }
+} /* namespace pywrap */