This commit is contained in:
skzhang1 2022-08-21 12:52:27 +00:00
parent bfacff2a3d
commit e3aa7ea9d1

View File

@ -541,7 +541,6 @@ def evaluate_model_CV(
total_fold_num += 1 total_fold_num += 1
val_loss_folds.append(val_loss_i) val_loss_folds.append(val_loss_i)
if log_training_metric or not isinstance(eval_metric, str): if log_training_metric or not isinstance(eval_metric, str):
if isinstance(metric_i, dict):
log_metric_folds.append(metric_i) log_metric_folds.append(metric_i)
train_time += train_time_i train_time += train_time_i
pred_time += pred_time_i pred_time += pred_time_i