Fix evaluation result for XGBRanker. (#6594)

* Remove duplicated code, which fixes typo `evals_result` -> `evals_result_`.
This commit is contained in:
Jiaming Yuan
2021-01-12 09:36:41 +08:00
committed by GitHub
parent f2f7dd87b8
commit c709f2aaaf
2 changed files with 11 additions and 21 deletions

View File

@@ -138,6 +138,7 @@ def test_ranking():
model = xgb.sklearn.XGBRanker(**params)
model.fit(x_train, y_train, group=train_group,
eval_set=[(x_valid, y_valid)], eval_group=[valid_group])
assert model.evals_result()
pred = model.predict(x_test)