From 19129b289c3b4833ae3e46bb173a1c63a953f3b0 Mon Sep 17 00:00:00 2001 From: Antonio Augusto Santos Date: Tue, 31 May 2016 19:01:10 -0300 Subject: [PATCH] Preserve the actal objective used on the booster Save the actual objective used on xgboost.train. Not saving it was giving problem in predict_proba, as issue #1215 --- python-package/xgboost/sklearn.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index 0a19d61c5..2b4c2accb 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -441,6 +441,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase): evals_result=evals_result, obj=obj, feval=feval, verbose_eval=verbose) + self.objective = xgb_options["objective"] if evals_result: for val in evals_result.items(): evals_result_key = list(val[1].keys())[0]