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

[python-package] Fix the issue #1439 (#1666)

Browse Source 
*Fix 1439
        *Fix python_wrapper when eval set name contain '-' will cause early_stop maximize variable con't set to True propely

Change-Id: Ib0595afd4ae7b445a84c00a3a8faeccc506c6d13
This commit is contained in:
ziguang1216 2016-10-19 01:22:51 +08:00 committed by Tianqi Chen
parent d3fc815b45
commit 94a9e3222e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/xgboost/callback.py
View File

@ -163,11 +163,11 @@ def early_stop(stopping_rounds, maximize=False, verbose=True):
maximize_score = maximize
metric = env.evaluation_result_list[-1][0]
if any(env.evaluation_result_list[-1][0].split('-')[1].startswith(x)
if any(env.evaluation_result_list[-1][0].split('-')[-1].startswith(x)
for x in maximize_at_n_metrics):
maximize_score = True
if any(env.evaluation_result_list[-1][0].split('-')[1].split(":")[0] == x
if any(env.evaluation_result_list[-1][0].split('-')[-1].split(":")[0] == x
for x in maximize_metrics):
maximize_score = True