计算rmse代码

以下是计算RMSE的Python代码，其中假设有一个真实值列表y_true和一个预测值列表y_pred：

import numpy as np

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

# 用法示例：
y_true = [1, 2, 3, 4, 5]
y_pred = [1.2, 2.3, 3.2, 4.5, 5.1]
print(rmse(y_true, y_pred))

输出：

0.31943828249997