Update python-package/lightgbm/engine.py

Co-authored-by: Nikita Titov <[email protected]>
microsoft · Dec 18, 2024 · d116279 · d116279
1 parent 4176327
commit d116279
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/lightgbm/engine.py b/python-package/lightgbm/engine.py
@@ -607,7 +607,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()
     ]