from sklearn.metrics import mean_squared_error mse = mean_squared_error(y_true, y_pred) rmse = mean_squared_error(y_true, y_pred, squared = False)