均方根误差

class pytorch_forecasting.metrics.point.RMSE(reduction='sqrt-mean', **kwargs)

基类: MultiHorizonMetric

均方根误差

定义为 (y_pred - target)**2

初始化指标

Parameters:

• name (str) – 指标名称。默认为类名称。

• 分位数 (列表[浮点数], 可选) – 概率范围的分位数。默认为 None。

• reduction (str, 可选) – 减少方式，“none”，“mean”或“sqrt-mean”。默认为“mean”。

方法

loss(y_pred, target)

计算未减少的损失。

loss(y_pred: Dict[str, Tensor], target)

计算不进行减小的损失。在派生类中覆盖

Parameters:

• y_pred – 网络输出

• y_actual – 实际值

Returns:

作为反向传播的单个数字的损失/指标

Return type:

torch.Tensor