fixing call to `mean_squared_error` to comply with the newer sklearn versions 3945a39 verified Nathan commited on Feb 17, 2025