均方误差#

mean_squared_error(y_true, y_pred, horizon_weight=None, multioutput='uniform_average', square_root=False, **kwargs)[源代码][源代码]#

均方误差 (MSE) 或均方根误差 (RMSE)。

如果 square_root 为 False，则计算均方误差（MSE）；如果 square_root 为 True，则计算均方根误差（RMSE）。MSE 和 RMSE 都是非负浮点数。最佳值为 0.0。

MSE 以输入数据的平方单位来衡量，而 RMSE 与数据的尺度相同。由于 MSE 和 RMSE 对预测误差进行平方而不是取绝对值，因此它们对大误差的惩罚比 MAE 更重。

参数:
y_truepd.Series, pd.DataFrame 或 np.array，形状为 (fh,) 或 (fh, n_outputs)，其中 fh 是预测范围

地面实况（正确）的目标值。

y_predpd.Series, pd.DataFrame 或 np.array，形状为 (fh,) 或 (fh, n_outputs)，其中 fh 是预测范围

预测值。

horizon_weight形状为 (fh,) 的类数组，默认为 None

预测范围权重。

多输出{‘raw_values’, ‘uniform_average’} 或形状为 (n_outputs,) 的类数组，默认=’uniform_average’

定义如何聚合多元（多输出）数据的度量。如果是类数组，则使用这些值作为权重来平均误差。如果是’raw_values’，则在多输出输入的情况下返回所有误差的完整集合。如果是’uniform_average’，则所有输出的误差以均匀权重平均。

平方根bool, 默认=False

是否取均方误差的平方根。如果为 True，则返回均方根误差（RMSE）；如果为 False，则返回均方误差（MSE）。

返回:
损失浮点数或浮点数的ndarray

MSE 损失。如果 multioutput 是 ‘raw_values’，则分别返回每个输出的 MSE。如果 multioutput 是 ‘uniform_average’ 或一个权重 ndarray，则返回所有输出误差的加权平均 MSE。

参见

mean_absolute_error
median_absolute_error
median_squared_error
geometric_mean_absolute_error
geometric_mean_squared_error

参考文献

Hyndman, R. J 和 Koehler, A. B. (2006)。“另一种看待预测准确度度量的方法”，《国际预测杂志》，第22卷，第4期。

示例

>>> from sktime.performance_metrics.forecasting import mean_squared_error
>>> y_true = np.array([3, -0.5, 2, 7, 2])
>>> y_pred = np.array([2.5, 0.0, 2, 8, 1.25])
>>> mean_squared_error(y_true, y_pred)
0.4125
>>> y_true = np.array([[0.5, 1], [-1, 1], [7, -6]])
>>> y_pred = np.array([[0, 2], [-1, 2], [8, -5]])
>>> mean_squared_error(y_true, y_pred)
0.7083333333333334
>>> mean_squared_error(y_true, y_pred, square_root=True)
0.8227486121839513
>>> mean_squared_error(y_true, y_pred, multioutput='raw_values')
array([0.41666667, 1.        ])
>>> mean_squared_error(y_true, y_pred, multioutput='raw_values', square_root=True)
array([0.64549722, 1.        ])
>>> mean_squared_error(y_true, y_pred, multioutput=[0.3, 0.7])
0.825
>>> mean_squared_error(y_true, y_pred, multioutput=[0.3, 0.7], square_root=True)
0.8936491673103708
On this page