牛客题解 | 计算均方根误差（RMSE）

题目

题目链接

均方根误差（RMSE）是一种衡量回归模型性能的指标，其计算公式为：

\[RMSE = \sqrt{\frac{1}{n} \sum_{i=1}^{n} (y_i - \hat{y}_i)^2} \]

其中，\(y_i\) 是真实值，\(\hat{y}_i\) 是预测值。

标准代码如下

def rmse(y_true, y_pred):
    return round(np.sqrt(np.mean((y_true - y_pred) ** 2)), 3)