From b63868327f90c26353bdd293c9183a303c684436 Mon Sep 17 00:00:00 2001 From: Eugene Nizhibitsky Date: Thu, 14 May 2015 22:55:49 +0300 Subject: [PATCH] Fix early stopping in python wrapper --- wrapper/xgboost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py index 5b38d2f2f..7e3fa2dc4 100644 --- a/wrapper/xgboost.py +++ b/wrapper/xgboost.py @@ -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: