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

Merge pull request #314 from enizhibitsky/wrapper_stopping_fix

Browse Source 
Fix early stopping in python wrapper
This commit is contained in:
Tianqi Chen 2015-05-14 16:16:47 -07:00
commit 9c52fc8e22
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: