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

Fix early stopping in python wrapper

Browse Source
This commit is contained in:
Eugene Nizhibitsky 2015-05-14 22:55:49 +03:00
parent 3a7808dc7d
commit b63868327f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wrapper/xgboost.py
View File

@ -657,7 +657,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
maximize_score = False
if 'eval_metric' in params:
maximize_metrics = ('auc', 'map', 'ndcg')
if filter(lambda x: params['eval_metric'].startswith(x), maximize_metrics):
if list(filter(lambda x: params['eval_metric'].startswith(x), maximize_metrics)):
maximize_score = True
if maximize_score: