如何衡量Python中最佳拟合线的质量?

时间:2014-09-24 22:43:54

标签: python numpy matplotlib scipy best-fit-curve

我有一些数据点要绘制,并希望在图表中添加最佳拟合线,然后输出相关指标以指示最佳拟合线的质量。

我可以绘制数据,polyfit是我用来添加最佳拟合线的函数。但是,我只是想知道如何获得指示最佳拟合线质量的指标?

我没有看到polyfit返回任何指标(即最小平方误差值)。

先谢谢。

数据:

0,1717
1,1761
2,1961
3,1711
4,1285
5,976
6,721
7,428
8,313
9,297
10,375
11,521
12,678
13,752
14,728
15,758
16,741
17,812
18,845
19,863
20,933
21,1169
22,1523
23,1779

1 个答案:

答案 0 :(得分:1)

均方误差是一种衡量标准。

如果你的线和数据上有每个点的numpy数组:

numpy.mean((data - line_vals) ** 2)

编辑: 获取line_vals,如果你有一个等式y = mx + b:

line_vals = b + m * numpy.linspace(0, 23, 24)