Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #1241 from KhaoticMind/master

Browse Source 
[py]Preserve the actual objective used on the booster - Fixed #1215
This commit is contained in:
Yuan (Terry) Tang 2016-06-01 09:11:49 -05:00
commit 9ef86072f4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
python-package/xgboost/sklearn.py
View File

@ -441,6 +441,7 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
evals_result=evals_result, obj=obj, feval=feval, evals_result=evals_result, obj=obj, feval=feval,
verbose_eval=verbose) verbose_eval=verbose)
self.objective = xgb_options["objective"]
if evals_result: if evals_result:
for val in evals_result.items(): for val in evals_result.items():
evals_result_key = list(val[1].keys())[0] evals_result_key = list(val[1].keys())[0]