fix cv bug (#269)

da0eb2bb · wxchan · Guolin Ke · 52177972 · da0eb2bb
Commit da0eb2bb authored Jan 27, 2017 by wxchan Committed by Guolin Ke Jan 27, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python-package/lightgbm/engine.py python-package/lightgbm/engine.py +1 -1

No files found.
--- a/python-package/lightgbm/engine.py
+++ b/python-package/lightgbm/engine.py
@@ -385,7 +385,7 @@ def cv(params, train_set, num_boost_round=10, nfold=5, stratified=False,
                                    begin_iteration=0,
                                    end_iteration=num_boost_round,
                                    evaluation_result_list=None))
-        cvfolds.update(fobj)
+        cvfolds.update(fobj=fobj)
        res = _agg_cv_result(cvfolds.eval_valid(feval))
        for _, key, mean, _, std in res:
            results[key + '-mean'].append(mean)