yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-02 02:40:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added an info reminding user that if no time_budget and no max_iter is specified, then effectively zero-shot AutoML is used

Browse Source
This commit is contained in:
Jing Dong 2022-12-13 14:19:09 -08:00
parent 232c356a4b
commit f6aa8eed3f
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
flaml/automl/automl.py
View File

@ -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