fix xgboost warning (#667)

mljar · Mar 4, 2024 · e38a8be · e38a8be
1 parent d9a1f8a
commit e38a8be
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/supervised/algorithms/xgboost.py b/supervised/algorithms/xgboost.py
@@ -201,7 +201,7 @@ def fit(
             early_stopping_rounds=esr,
             evals_result=evals_result,
             verbose_eval=False,
-            feval=self.custom_eval_metric
+            custom_metric=self.custom_eval_metric
             # callbacks=[time_constraint] # callback slows down by factor ~8
         )
 

diff --git a/supervised/tuner/optuna/xgboost.py b/supervised/tuner/optuna/xgboost.py
@@ -112,7 +112,7 @@ def __call__(self, trial):
                 early_stopping_rounds=self.early_stopping_rounds,
                 callbacks=[pruning_callback],
                 verbose_eval=False,
-                feval=self.custom_eval_metric,
+                custom_metric=self.custom_eval_metric,
             )
             preds = bst.predict(
                 self.dvalidation, iteration_range=(0, bst.best_iteration)