tensorflow tf.keras.losses.MeanSquaredError 均方差损失函数 示例

import tensorflow as tf
import numpy as np

公式

$error=\frac{1}{m}{\sum }_{i=1}^m(x_i-y_i{)}^2$

y_true = np.array([[1., 2.], [3., 4.]])
y_pred = np.array([[5., 6.], [9., 10.]])
mse = tf.keras.losses.MeanSquaredError()  # 均方差损失函数
mse(y_true, y_pred).numpy()

26.0

计算过程

$\frac{1}{4}\times [(1-5{)}^{{}^2}+(2-6{)}^2+(3-9{)}^2+(4-10{)}^2]=26$

mse = tf.keras.losses.MeanSquaredError(
    reduction=tf.keras.losses.Reduction.NONE # 分开计算损失值
)  # 均方差损失函数
mse(y_true, y_pred).numpy()

array([16., 36.])

计算过程

$\frac{1}{2}\times [(1-5{)}^{{}^2}+(2-6{)}^2],\frac{1}{2}\times [(3-9{)}^2+(4-10{)}^2]=[16,36]$​