Skip to content

Commit e38a8be

Browse files
committed
fix xgboost warning (#667)
1 parent d9a1f8a commit e38a8be

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

supervised/algorithms/xgboost.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ def fit(
201201
early_stopping_rounds=esr,
202202
evals_result=evals_result,
203203
verbose_eval=False,
204-
feval=self.custom_eval_metric
204+
custom_metric=self.custom_eval_metric
205205
# callbacks=[time_constraint] # callback slows down by factor ~8
206206
)
207207

supervised/tuner/optuna/xgboost.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def __call__(self, trial):
112112
early_stopping_rounds=self.early_stopping_rounds,
113113
callbacks=[pruning_callback],
114114
verbose_eval=False,
115-
feval=self.custom_eval_metric,
115+
custom_metric=self.custom_eval_metric,
116116
)
117117
preds = bst.predict(
118118
self.dvalidation, iteration_range=(0, bst.best_iteration)

0 commit comments

Comments
 (0)