From 3b158e9fc1eceae30b11cd525e6237b543aa8830 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Sun, 15 Jun 2025 12:07:12 +0200 Subject: [PATCH] Add AdaBoost --- src/main/Models.h | 1 + src/main/modelRegister.h | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main/Models.h b/src/main/Models.h index 565a96d..076a926 100644 --- a/src/main/Models.h +++ b/src/main/Models.h @@ -23,6 +23,7 @@ #include #include #include +#include #include #include "../experimental_clfs/XA1DE.h" diff --git a/src/main/modelRegister.h b/src/main/modelRegister.h index 0dbd269..22a800c 100644 --- a/src/main/modelRegister.h +++ b/src/main/modelRegister.h @@ -35,6 +35,8 @@ namespace platform { [](void) -> bayesnet::BaseClassifier* { return new pywrap::RandomForest();}); static Registrar registrarXGB("XGBoost", [](void) -> bayesnet::BaseClassifier* { return new pywrap::XGBoost();}); + static Registrar registrarAda("AdaBoost", + [](void) -> bayesnet::BaseClassifier* { return new pywrap::AdaBoost();}); static Registrar registrarXSPODE("XSPODE", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::XSpode(0);}); static Registrar registrarXSP2DE("XSP2DE", @@ -44,6 +46,6 @@ namespace platform { static Registrar registrarXBA2DE("XBA2DE", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::XBA2DE();}); static Registrar registrarXA1DE("XA1DE", - [](void) -> bayesnet::BaseClassifier* { return new XA1DE();}); + [](void) -> bayesnet::BaseClassifier* { return new XA1DE();}); } #endif