Update flaml/automl.py

Co-authored-by: Chi Wang <wang.chi@microsoft.com>
This commit is contained in:
zsk 2022-08-21 08:35:44 -04:00 committed by GitHub
parent 5faafa01d4
commit 39d8c0e3e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -2376,7 +2376,7 @@ class AutoML(BaseEstimator):
def cv_score_agg_func(val_loss_folds, log_metrics_folds): def cv_score_agg_func(val_loss_folds, log_metrics_folds):
return metric_to_minimize, metrics_to_log 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. 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., E.g.,