From 3192bf82d861dd55cef8a2c0e8cd71e104cf90e3 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Wed, 3 Sep 2014 12:15:57 -0700 Subject: [PATCH] Update xgboost.py --- wrapper/xgboost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py index e09c6da8c..5b4eee6b8 100644 --- a/wrapper/xgboost.py +++ b/wrapper/xgboost.py @@ -448,7 +448,7 @@ def mknfold(dall, nfold, param, seed, evals=[], fpreproc = None): # run preprocessing on the data set if needed if fpreproc is not None: dtrain, dtest, tparam = fpreproc(dtrain, dtest, param.copy()) - plst = param.items() + [('eval_metric', itm) for itm in evals] + plst = tparam.items() + [('eval_metric', itm) for itm in evals] ret.append(CVPack(dtrain, dtest, plst)) return ret