numpy矩阵求MSE

MSE loss

#官方示例
from sklearn.metrics import mean_squared_error
y_true = [[0.5, 1],[-1, 1],[7, -6]]
y_pred = [[0, 2],[-1, 2],[8, -5]]
mean_squared_error(y_true, y_pred) #0.708

验证

import numpy as np 
A = np.array(y_true)
B = np.array(y_pred)

mse = (np.square(A - B)).mean(axis=None)
mse 

结果如下

总结如下

#官方示例
from sklearn.metrics import mean_squared_error

import numpy as np 
y_true = [[0.5, 1],[-1, 1],[7, -6]]
y_pred = [[0, 2],[-1, 2],[8, -5]]
mse=mean_squared_error(y_true, y_pred) #0.708
# (0.25+1+0+1+1+1)/(2*3)
print(mse)
print((0.25+1+0+1+1+1)/(2*3))

A = np.array(y_true)
B = np.array(y_pred)

mse2 = (np.square(A - B)).mean(axis=None)
print(mse2) 

mse3 = np.mean((A - B)**2)
print(mse3)

结果如下