From d31e14617e45216b49175f02767b3161433f0e7d Mon Sep 17 00:00:00 2001 From: Jing Dong <35925336+jingdong00@users.noreply.github.com> Date: Mon, 19 Dec 2022 01:49:00 +0800 Subject: [PATCH] Added an info reminding user that if no time_budget and no max_iter is specified, then effectively zero-shot AutoML is used (#850) * Added an info reminding user that if no time_budget and no max_iter is specified, then effectively zero-shot AutoML is used * moved message to line 2818 * Update flaml/automl/automl.py Co-authored-by: Chi Wang * Update flaml/automl/automl.py Co-authored-by: Chi Wang Co-authored-by: Chi Wang Co-authored-by: Qingyun Wu --- flaml/automl/automl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/flaml/automl/automl.py b/flaml/automl/automl.py index 078ce67a2091..db092b524f4d 100644 --- a/flaml/automl/automl.py +++ b/flaml/automl/automl.py @@ -2566,6 +2566,7 @@ def cv_score_agg_func(val_loss_folds, log_metrics_folds): self._use_ray = use_ray or n_concurrent_trials > 1 # use the following condition if we have an estimation of average_trial_time and average_trial_overhead # self._use_ray = use_ray or n_concurrent_trials > ( average_trail_time + average_trial_overhead) / (average_trial_time) + if self._use_ray is not False: import ray @@ -2816,6 +2817,7 @@ def is_to_reverse_metric(metric, task): logger.warning( "No search budget is provided via time_budget or max_iter." " Training only one model per estimator." + " Zero-shot AutoML is used for certain tasks and estimators." " To tune hyperparameters for each estimator," " please provide budget either via time_budget or max_iter." )