Fix evaluation result for XGBRanker. (#6594) (#6600)

* Remove duplicated code, which fixes typo `evals_result` -> `evals_result_`.
This commit is contained in:
Jiaming Yuan
2021-01-13 04:42:43 +08:00
committed by GitHub
parent 8e321adac8
commit 6a29afb480
2 changed files with 13 additions and 18 deletions

View File

@@ -122,6 +122,8 @@ 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)
train_data = xgb.DMatrix(x_train, y_train)