Update xgboost.py

2014-09-03 12:15:57 -07:00 · 2014-09-03 12:15:57 -07:00 · 3192bf82d8
commit 3192bf82d8
parent 0c36231ea3
1 changed files with 1 additions and 1 deletions
--- 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