mean_squared_error()

Среднеквадратичная ошибка (MSE / RMSE).

Сигнатураmean_squared_error(y_true, y_pred, squared=True)

Среднее значение квадратов отклонений предсказаний от истины. Сильно штрафует крупные ошибки. Для RMSE возьмите корень или передайте squared=False (в новых версиях используйте root_mean_squared_error).

from sklearn.metrics import mean_squared_error
import numpy as np

mse = mean_squared_error(y_test, y_pred)
print('RMSE:', np.sqrt(mse))

← Все записи: scikit-learn