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

Fix how maximize_metric value is determined in early_stop (#1451)

Browse Source 
* Update callback.py

* Update callback.py
This commit is contained in:
kiselev1189 2016-08-28 00:09:24 +04:00 committed by Tianqi Chen
parent df38f251be
commit 53ce511be3
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -159,9 +159,15 @@ def early_stop(stopping_rounds, maximize=False, verbose=True):
"'{0}' will be used for early stopping.\n\n")
rabit.tracker_print(msg.format(env.evaluation_result_list[-1][0]))
maximize_metrics = ('auc', 'map', 'ndcg')
maximize_at_n_metrics = ('auc@', 'map@', 'ndcg@')
maximize_score = maximize
metric = env.evaluation_result_list[-1][0]
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
for x in maximize_metrics):
maximize_score = True