From 79948586978a5d9736e8c1372d0f608294476e3e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Zygmunt=20Zaj=C4=85c?= Date: Mon, 30 Mar 2015 19:58:25 +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 5a5d59b11..a2ecacbe2 100644 --- a/wrapper/xgboost.py +++ b/wrapper/xgboost.py @@ -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 maximize_score = False if 'eval_metric' in params: - maximize_metrics = ('auc', 'map', 'ndcg') - if filter( lambda x: params['eval_metric'].startswith(x), maximize_metrics ): + maximize_metrics = ('auc', 'map', 'ndcg') + if filter( lambda x: params['eval_metric'].startswith(x), maximize_metrics ): maximize_score = True if maximize_score: