Spaces:
Sleeping
Sleeping
update call to `mean_squared_error` to comply with sklearn v1.6
#4
by
NathanFradet
- opened
The squared argument of the scimitar-learn method mean_squared_error is depreciated since v1.4 and has been removed in v1.6, making this evaluation module incompatible with the latest version.
This PR makes it call the root_mean_squared_error when squared=True, solving this error.
For reference: https://scikit-learn.org/1.5/modules/generated/sklearn.metrics.mean_squared_error.html (docs of v1.5)
cc @lvwerra