python中求rmse

在Python中，可以使用NumPy库来计算均方根误差（RMSE）。假设有两个数组y_true和y_pred，分别表示真实值和预测值，可以按照以下方式计算RMSE：

import numpy as np

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

其中，np.mean()函数用于计算数组元素的平均值，**表示平方，np.sqrt()函数用于计算平方根。**

使用示例：

y_true = np.array([1, 2, 3])
y_pred = np.array([1.2, 2.1, 2.8])
print(rmse(y_true, y_pred))

输出结果为：

0.28284271247461906