yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-09-27 17:20:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
skzhang1 2022-08-21 17:44:26 +00:00
parent 462c27f8ae
commit 50fb20ebbc
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
flaml/ml.py
View File

@ -540,8 +540,7 @@ def evaluate_model_CV(
fit_kwargs["sample_weight"] = weight fit_kwargs["sample_weight"] = weight
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): 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
if time.time() - start_time >= budget: if time.time() - start_time >= budget: