From 5d2929cb83d5057cd912c1c7c1f562cd46ae9679 Mon Sep 17 00:00:00 2001 From: zsk Date: Sun, 21 Aug 2022 08:47:20 -0400 Subject: [PATCH] Update flaml/automl.py Co-authored-by: Chi Wang --- flaml/automl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flaml/automl.py b/flaml/automl.py index 0006b2be0..2a0c20059 100644 --- a/flaml/automl.py +++ b/flaml/automl.py @@ -2393,7 +2393,7 @@ class AutoML(BaseEstimator): metrics_to_log += single_fold if metrics_to_log: n = len(val_loss_folds) - metrics_to_log = {k: v / n for k, v in metrics_to_log.items()} if isinstance(metrics_to_log, dict) else metrics_to_log/n + metrics_to_log = {k: v / n for k, v in metrics_to_log.items()} if isinstance(metrics_to_log, dict) else metrics_to_log / n return metric_to_minimize, metrics_to_log ```