Keep searching hyperparameters when `r2_score` raises an error #1325

Atry · 2024-07-27T04:13:04Z

Why are these changes needed?

r2_score may raise an error when the model output is an inf or nan value. This could happen whening training MLP models

I've included any doc changes needed for https://microsoft.github.io/FLAML/. See https://microsoft.github.io/FLAML/docs/Contribute#documentation to build and test documentation locally.
I've added tests (if relevant) corresponding to the changes introduced in this PR.
I've made sure all auto checks have passed.

Atry · 2024-07-27T04:16:26Z

@microsoft-github-policy-service agree [company="ProtonBase"]

Atry · 2024-07-27T04:17:01Z

@microsoft-github-policy-service agree

flaml/automl/ml.py

Keep searching hyperparameters when r2_score raises an error

9635cd6

Atry marked this pull request as ready for review July 28, 2024 18:19

thinkall reviewed Aug 1, 2024

View reviewed changes

flaml/automl/ml.py Show resolved Hide resolved

Merge branch 'main' into patch-2

0218158

thinkall approved these changes Aug 6, 2024

View reviewed changes

flaml/automl/ml.py Outdated Show resolved Hide resolved

thinkall added 2 commits August 6, 2024 21:23

Add log info

3c810f4

Merge branch 'main' into patch-2

709bb13

thinkall enabled auto-merge August 6, 2024 13:25

thinkall added this pull request to the merge queue Aug 6, 2024

Merged via the queue into microsoft:main with commit 853c950 Aug 6, 2024
16 checks passed