From 50fb20ebbcc2726d5868f5441e86f01f5bd4bbb6 Mon Sep 17 00:00:00 2001 From: skzhang1 Date: Sun, 21 Aug 2022 17:44:26 +0000 Subject: [PATCH] update --- flaml/ml.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/flaml/ml.py b/flaml/ml.py index 044424801..1d50512fb 100644 --- a/flaml/ml.py +++ b/flaml/ml.py @@ -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: