首页 > 解决方案 > 如何在没有 numpy 的情况下计算 RMSE?

问题描述

刚刚为工作做了一个编码挑战。

一项任务是计算预测值和观测值之间的均方根误差。

预料到的:

[4, 25, 0.75, 11]

观察到的:

[3, 21, -1.25, 13]

结果将是 2.5。

numpy 不可用。我没有完成这项任务,但我想知道如何使用纯 Python 3 做到这一点?

标签: pythonstatistics

解决方案


这是我的做法:

pred = [4, 25, 0.75, 11]
observed = [3, 21, -1.25, 13]
error = [(p - o) for p, o in zip(pred, observed)]
square_error = [e**2 for e in error]
mean_square_error = sum(square_error)/len(square_error)
root_mean_square_error = mean_square_error**0.5

推荐阅读