From 7e7a9299fc040b8748ecda656c87b11567be9470 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Thu, 18 Jan 2024 10:23:24 +0100 Subject: [PATCH] Make xgboost version same a sklearn --- src/PyClassifiers/XGBoost.cc | 6 +----- src/PyClassifiers/XGBoost.h | 1 - 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/src/PyClassifiers/XGBoost.cc b/src/PyClassifiers/XGBoost.cc index a073b1a..e1bcfb6 100644 --- a/src/PyClassifiers/XGBoost.cc +++ b/src/PyClassifiers/XGBoost.cc @@ -2,12 +2,8 @@ //See https ://stackoverflow.com/questions/36071672/using-xgboost-in-c namespace pywrap { - XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier") + XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier", true) { validHyperparameters = { "tree_method", "early_stopping_rounds", "n_jobs" }; } - std::string XGBoost::version() - { - return callMethodString("1.0"); - } } /* namespace pywrap */ \ No newline at end of file diff --git a/src/PyClassifiers/XGBoost.h b/src/PyClassifiers/XGBoost.h index b384ded..930b92e 100644 --- a/src/PyClassifiers/XGBoost.h +++ b/src/PyClassifiers/XGBoost.h @@ -7,7 +7,6 @@ namespace pywrap { public: XGBoost(); ~XGBoost() = default; - std::string version(); }; } /* namespace pywrap */ #endif /* XGBOOST_H */ \ No newline at end of file