This commit is contained in:
skzhang1 2022-08-21 17:44:26 +00:00
parent 462c27f8ae
commit 50fb20ebbc

View File

@ -540,8 +540,7 @@ def evaluate_model_CV(
fit_kwargs["sample_weight"] = weight
total_fold_num += 1
val_loss_folds.append(val_loss_i)
if log_training_metric or not isinstance(eval_metric, str):
log_metric_folds.append(metric_i)
log_metric_folds.append(metric_i)
train_time += train_time_i
pred_time += pred_time_i
if time.time() - start_time >= budget: