Skip to content

Commit

Permalink
Use old objective.
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Feb 28, 2023
1 parent 8700297 commit d553e7c
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/python/test_eval_metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -299,7 +299,9 @@ def test_pr_auc_multi(self):
def run_pr_auc_ltr(self, tree_method):
from sklearn.datasets import make_classification
X, y = make_classification(128, 4, n_classes=2, random_state=1994)
ltr = xgb.XGBRanker(tree_method=tree_method, n_estimators=16)
ltr = xgb.XGBRanker(
tree_method=tree_method, n_estimators=16, objective="rank:pairwise"
)
groups = np.array([32, 32, 64])
ltr.fit(
X,
Expand Down

0 comments on commit d553e7c

Please sign in to comment.