From f6aa8eed3f30ac3c0207efa9c07fd15a2a76da44 Mon Sep 17 00:00:00 2001 From: Jing Dong Date: Tue, 13 Dec 2022 14:19:09 -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 --- flaml/automl/automl.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/flaml/automl/automl.py b/flaml/automl/automl.py index 078ce67a2..e5eecb3fa 100644 --- a/flaml/automl/automl.py +++ b/flaml/automl/automl.py @@ -2566,6 +2566,11 @@ class AutoML(BaseEstimator): 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 no time_budget and no max_iter is specified, then effectively zero-shot AutoML is used. + if time_budget is -1 and max_iter is None: + logger.info('Neither time_budegt nor max_iter is specified, zero-shot ML is used. ') + if self._use_ray is not False: import ray