From 39093bc432e5981f2c11b53617690a5214ad22dd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Zygmunt=20Zaj=C4=85c?= Date: Mon, 30 Mar 2015 19:59:09 +0200 Subject: [PATCH] early stopping for Python wrapper --- wrapper/xgboost.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py index a2ecacbe2..da891ab6c 100644 --- a/wrapper/xgboost.py +++ b/wrapper/xgboost.py @@ -578,7 +578,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 filter(lambda x: params['eval_metric'].startswith(x), maximize_metrics): maximize_score = True if maximize_score: @@ -601,7 +601,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None): sys.stderr.write(msg + '\n') - score = float(msg.rsplit( ':', 1 )[1]) + score = float(msg.rsplit(':', 1)[1]) if (maximize_score and score > best_score) or \ (not maximize_score and score < best_score): best_score = score