Skip to content

fix(mse): support sklearn versions without root_mean_squared_error - #805

Open
ShamikOfficial wants to merge 1 commit into
huggingface:mainfrom
ShamikOfficial:fix-mse-sklearn-compat-669
Open

fix(mse): support sklearn versions without root_mean_squared_error#805
ShamikOfficial wants to merge 1 commit into
huggingface:mainfrom
ShamikOfficial:fix-mse-sklearn-compat-669

Conversation

@ShamikOfficial

Copy link
Copy Markdown

Fixes #669

Summary

  • Add sklearn import fallback when root_mean_squared_error is unavailable
  • Fix RMSE README example label (squared=False)

Test plan

  • Verified MSE/RMSE compute paths for both sklearn code paths

Add a fallback for older sklearn releases and fix the RMSE README example label.

Fixes huggingface#669
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

TypeError: got an unexpected keyword argument 'squared'

1 participant