Spaces:
Running
Running
update call to `mean_squared_error` to comply with sklearn v1.6
#4
by
Natooz
- 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