python-3.x - 如何使用 numpy 在 python 中计算 RMSPE

Question

我正在使用 Rossmann dataset. 进行多元预测我现在需要使用 RMSPE 指标来评估我的模型。我看到了相关公式here 。但我不确定如何使用 numpy 有效地实现这一点。任何帮助深表感谢。

Answer 1

您可以利用 numpy 的矢量化功能来实现这样的错误度量。以下函数可用于计算 RMSPE:

def rmse(y_true, y_pred):
    '''
    Compute Root Mean Square Percentage Error between two arrays.
    '''
    loss = np.sqrt(np.mean(np.square(((y_true - y_pred) / y_true)), axis=0))

    return loss

(对于向量之间的误差，axis=0 明确表示误差是按行计算的，返回一个向量。这不是必需的，因为这是 np.mean。)