rmontanana/Pyclassifiers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add AdaBoostPy

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2025-06-18 21:31:36 +02:00
parent 2a99dce23b
commit e460eb4c41
5 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pyclfs/AdaBoost.cc → pyclfs/AdaBoostPy.cc
View File

@@ -1,19 +1,19 @@
#include "AdaBoost.h"
#include "AdaBoostPy.h"
namespace pywrap {
AdaBoost::AdaBoost() : PyClassifier("sklearn.ensemble", "AdaBoostClassifier", true)
AdaBoostPy::AdaBoostPy() : PyClassifier("sklearn.ensemble", "AdaBoostClassifier", true)
{
validHyperparameters = { "n_estimators", "n_jobs", "random_state" };
}
int AdaBoost::getNumberOfEdges() const
int AdaBoostPy::getNumberOfEdges() const
{
return callMethodSumOfItems("get_n_leaves");
}
int AdaBoost::getNumberOfStates() const
int AdaBoostPy::getNumberOfStates() const
{
return callMethodSumOfItems("get_depth");
}
int AdaBoost::getNumberOfNodes() const
int AdaBoostPy::getNumberOfNodes() const
{
return callMethodSumOfItems("node_count");
}

10
pyclfs/AdaBoost.h → pyclfs/AdaBoostPy.h
View File

@@ -1,12 +1,12 @@
#ifndef ADABOOST_H
#define ADABOOST_H
#ifndef ADABOOSTPY_H
#define ADABOOSTPY_H
#include "PyClassifier.h"
namespace pywrap {
class AdaBoost : public PyClassifier {
class AdaBoostPy : public PyClassifier {
public:
AdaBoost();
~AdaBoost() = default;
AdaBoostPy();
~AdaBoostPy() = default;
int getNumberOfEdges() const override;
int getNumberOfStates() const override;
int getNumberOfNodes() const override;

2
pyclfs/CMakeLists.txt
View File

@@ -4,5 +4,5 @@ include_directories(
${PyClassifiers_SOURCE_DIR}/lib/json/include
${Bayesnet_INCLUDE_DIRS}
)
add_library(PyClassifiers ODTE.cc STree.cc SVC.cc RandomForest.cc XGBoost.cc AdaBoost.cc PyClassifier.cc PyWrap.cc)
add_library(PyClassifiers ODTE.cc STree.cc SVC.cc RandomForest.cc XGBoost.cc AdaBoostPy.cc PyClassifier.cc PyWrap.cc)
target_link_libraries(PyClassifiers nlohmann_json::nlohmann_json ${Python3_LIBRARIES} "${TORCH_LIBRARIES}" ${LIBTORCH_PYTHON} Boost::boost Boost::python Boost::numpy)