Add AdaBoostPy

2025-06-18 21:31:36 +02:00
parent 2a99dce23b
commit e460eb4c41
5 changed files with 14 additions and 14 deletions
--- a/pyclfs/AdaBoostPy.cc
+++ b/pyclfs/AdaBoostPy.cc
@@ -0,0 +1,20 @@
+#include "AdaBoostPy.h"
+
+namespace pywrap {
+    AdaBoostPy::AdaBoostPy() : PyClassifier("sklearn.ensemble", "AdaBoostClassifier", true)
+    {
+        validHyperparameters = { "n_estimators", "n_jobs", "random_state" };
+    }
+    int AdaBoostPy::getNumberOfEdges() const
+    {
+        return callMethodSumOfItems("get_n_leaves");
+    }
+    int AdaBoostPy::getNumberOfStates() const
+    {
+        return callMethodSumOfItems("get_depth");
+    }
+    int AdaBoostPy::getNumberOfNodes() const
+    {
+        return callMethodSumOfItems("node_count");
+    }
+} /* namespace pywrap */