python - Python 中的均方误差

Question

我正在尝试制作一个函数来计算 y(真实值)和 y_pred(预测值)的均方误差，而不是使用 sklearn 或其他实现。

接下来我会尝试:

def mserror(y, y_pred):
    i=0
    for i in range (len(y)):
        i+=1
        mse = ((y - y_pred) ** 2).mean(y)   
        return mse

能否请您纠正我在计算中做错了什么，谁可以解决这个问题？

Answer 1

您正在无缘无故地修改索引。无论如何，for 循环都会增加它。此外，您没有使用索引，例如，您没有使用任何 y[i] - y_pred[i]，因此您根本不需要循环。

使用数组

mse = np.mean((y - y_pred)**2)