Added an info reminding user that if no time_budget and no max_iter i…

…s specified, then effectively zero-shot AutoML is used (microsoft#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 <[email protected]> * Update flaml/automl/automl.py Co-authored-by: Chi Wang <[email protected]> Co-authored-by: Chi Wang <[email protected]> Co-authored-by: Qingyun Wu <[email protected]>
whiskyboy · Dec 18, 2022 · d31e146 · d31e146
1 parent eb417f7
commit d31e146
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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."
             )