Fix xgboost error in predict/predict_proba

2025-04-12 17:48:23 +02:00
parent 761f57be6c
commit 830265d91b
4 changed files with 55 additions and 30 deletions
--- a/pyclfs/XGBoost.cc
+++ b/pyclfs/XGBoost.cc
@@ -5,5 +5,6 @@ namespace pywrap {
    XGBoost::XGBoost() : PyClassifier("xgboost", "XGBClassifier", true)
    {
        validHyperparameters = { "tree_method", "early_stopping_rounds", "n_jobs" };
+        xgboost = true;
    }
 } /* namespace pywrap */