yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-01 10:19:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update flaml/automl.py

Browse Source 
Co-authored-by: Chi Wang <wang.chi@microsoft.com>
This commit is contained in:
zsk 2022-08-21 08:47:20 -04:00 committed by GitHub
parent 39d8c0e3e8
commit 5d2929cb83
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flaml/automl.py
View File

@ -2393,7 +2393,7 @@ class AutoML(BaseEstimator):
metrics_to_log += single_fold
if metrics_to_log:
n = len(val_loss_folds)
metrics_to_log = {k: v / n for k, v in metrics_to_log.items()} if isinstance(metrics_to_log, dict) else metrics_to_log/n
metrics_to_log = {k: v / n for k, v in metrics_to_log.items()} if isinstance(metrics_to_log, dict) else metrics_to_log / n
return metric_to_minimize, metrics_to_log
```