From 42200ec03ec6f2593c2a930bd9f3ce38fa11be5e Mon Sep 17 00:00:00 2001 From: Rory Mitchell Date: Thu, 1 Nov 2018 13:34:25 +1300 Subject: [PATCH] Allow XGBRanker sklearn interface to use other xgboost ranking objectives (#3848) --- python-package/xgboost/sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index dcc4e85a9..1f373687e 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -921,7 +921,7 @@ class XGBRanker(XGBModel): base_score, random_state, seed, missing) if callable(self.objective): raise ValueError("custom objective function not supported by XGBRanker") - elif self.objective != "rank:pairwise": + elif "rank:" not in self.objective: raise ValueError("please use XGBRanker for ranking task") def fit(self, X, y, group, sample_weight=None, eval_set=None, sample_weight_eval_set=None,