diff --git a/python-package/lightgbm/engine.py b/python-package/lightgbm/engine.py index 64f44751b5d3..cca29034b39c 100644 --- a/python-package/lightgbm/engine.py +++ b/python-package/lightgbm/engine.py @@ -607,7 +607,7 @@ def _agg_cv_result( # (, , mean(), , std_dev()) # ] return [ - (k[0], k[1], float(np.mean(metric_values[k])), metric_types[k], float(np.std(metric_values[k]))) + (k[0], k[1], float(np.mean(v)), metric_types[k], float(np.std(v))) for k, v in metric_values.items() ]