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

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment