yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-27 06:59:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
skzhang1 2022-08-24 02:58:18 +00:00
parent 50fb20ebbc
commit 96f0688595
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flaml/ml.py
View File

@ -536,6 +536,8 @@ def evaluate_model_CV(
log_training_metric=log_training_metric,
fit_kwargs=fit_kwargs,
)
if isinstance(metric_i, dict) and "intermediate_results" in metric_i.keys():
del metric_i["intermediate_results"]
if weight is not None:
fit_kwargs["sample_weight"] = weight
total_fold_num += 1