From 39d8c0e3e8140b50298883aa1dec7c34f0db27d0 Mon Sep 17 00:00:00 2001 From: zsk Date: Sun, 21 Aug 2022 08:35:44 -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 70f8ed4bf..0006b2be0 100644 --- a/flaml/automl.py +++ b/flaml/automl.py @@ -2376,7 +2376,7 @@ class AutoML(BaseEstimator): def cv_score_agg_func(val_loss_folds, log_metrics_folds): return metric_to_minimize, metrics_to_log ``` - “val_loss_folds” - list of float, it records the loss scores of each ford; “log_metrics_folds” - list of dict/float, it records the metrics of each fords to log. + “val_loss_folds” - list of floats, the loss scores of each fold; “log_metrics_folds” - list of dicts/floats, the metrics of each fold to log. This function should return the final aggregate result of all folds. A float number of the minimization objective, and a dictionary as the metrics to log or None. E.g.,