From d116279bb571f309f011f6d67824866590c4f300 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Tue, 17 Dec 2024 23:26:43 -0600 Subject: [PATCH] Update python-package/lightgbm/engine.py Co-authored-by: Nikita Titov --- python-package/lightgbm/engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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() ]