牛客题解 | 计算均方根误差（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)