Fix sklearn deprecation warning for parameter squared of function mean_squared_error

aimclub / BAMT

Repository of a data modeling and analysis tool based on Bayesian networks

https://bamt.readthedocs.io

BSD 3-Clause "New" or "Revised" License

118 stars 17 forks source link

Fix sklearn deprecation warning for parameter squared of function mean_squared_error #114

Open anton-golubkov opened 3 months ago

anton-golubkov commented 3 months ago

Calling mean_squared_error(..., squared=False) shows deprecation warning in sckit-learn 1.4: Deprecated since version 1.4: squared is deprecated in 1.4 and will be removed in 1.6. Use [root_mean_squared_error] instead to calculate the root mean squared error.

This commit replaces mean_squared_error with root_mean_squared_error to fix this warning.

jrzkaminski commented 3 months ago

A decent improvement, thank you, I'll merge it asap after tests pass