Merge branch 'python/remove-cv-agg' of github.com:microsoft/LightGBM …

…into python/remove-cv-agg
microsoft · Dec 18, 2024 · c5a69fd · c5a69fd
2 parents 47bd9d7 + d116279
commit c5a69fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/lightgbm/engine.py b/python-package/lightgbm/engine.py
@@ -606,7 +606,7 @@ def _agg_cv_result(
     #     (<dataset_name>, <metric_name>, mean(<values>), <is_higher_better>, std_dev(<values>))
     # ]
     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()
     ]