fix: Cross validation process isn't always run to completion (#1360)

microsoft · Oct 1, 2024 · 5c0f18b · 5c0f18b
1 parent e5d95f5
commit 5c0f18b
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/flaml/automl/task/generic_task.py b/flaml/automl/task/generic_task.py
@@ -706,7 +706,6 @@ def evaluate_model_CV(
  fit_kwargs = {}
  if cv_score_agg_func is None:
  cv_score_agg_func = default_cv_score_agg_func
- start_time = time.time()
  val_loss_folds = []
  log_metric_folds = []
  metric = None
@@ -813,8 +812,6 @@ def evaluate_model_CV(
  if is_spark_dataframe:
  X_train.spark.unpersist() # uncache data to free memory
  X_val.spark.unpersist() # uncache data to free memory
- if budget and time.time() - start_time >= budget:
- break
  val_loss, metric = cv_score_agg_func(val_loss_folds, log_metric_folds)
  n = total_fold_num
  pred_time /= n