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