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

early stopping for Python wrapper

Browse Source
This commit is contained in:
Zygmunt Zając 2015-03-30 19:58:25 +02:00
parent f9e157011f
commit 7994858697
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wrapper/xgboost.py
View File

@ -577,8 +577,8 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None):
# either minimize loss or maximize AUC/MAP/NDCG # either minimize loss or maximize AUC/MAP/NDCG
maximize_score = False maximize_score = False
if 'eval_metric' in params: if 'eval_metric' in params:
maximize_metrics = ('auc', 'map', 'ndcg') maximize_metrics = ('auc', 'map', 'ndcg')
if filter( lambda x: params['eval_metric'].startswith(x), maximize_metrics ): if filter( lambda x: params['eval_metric'].startswith(x), maximize_metrics ):
maximize_score = True maximize_score = True
if maximize_score: if maximize_score: