21

如何在 python 中计算最佳拟合线,然后将其绘制在 matplotlib 的散点图上?

我是我使用普通最小二乘回归计算线性最佳拟合线,如下所示:

from sklearn import linear_model
clf = linear_model.LinearRegression()
x = [[t.x1,t.x2,t.x3,t.x4,t.x5] for t in self.trainingTexts]
y = [t.human_rating for t in self.trainingTexts]
clf.fit(x,y)
regress_coefs = clf.coef_
regress_intercept = clf.intercept_      

这是多变量的(每种情况都有许多 x 值)。因此,X 是列表列表,y 是单个列表。例如:

x = [[1,2,3,4,5], [2,2,4,4,5], [2,2,4,4,1]] 
y = [1,2,3,4,5]

但是如何使用高阶多项式函数来做到这一点。例如,不仅是线性的(x 的 M=1 次方),还有二项式(x 的 M=2 次方)、二次方(x 的 M=4 次方)等等。例如,如何从以下获得最佳拟合曲线?

摘自 Christopher Bishops 的《模式识别与机器学习》,第 7 页:

摘自克里斯托弗·毕晓普斯的

4

2 回答 2

28

这个问题的公认答案 提供了一个小的多多边形拟合库,它将完全满足您使用 numpy 的需要,您可以将结果插入到绘图中,如下所述。

您只需将 x 和 y 点数组以及所需的拟合度(顺序)传入multipolyfit. 这将返回您可以使用 numpy 的 polyval 进行绘图的系数。

注意:下面的代码已被修改为进行多元拟合,但绘图图像是早期非多元答案的一部分。

import numpy
import matplotlib.pyplot as plt
import multipolyfit as mpf

data = [[1,1],[4,3],[8,3],[11,4],[10,7],[15,11],[16,12]]
x, y = zip(*data)
plt.plot(x, y, 'kx')

stacked_x = numpy.array([x,x+1,x-1])
coeffs = mpf(stacked_x, y, deg) 
x2 = numpy.arange(min(x)-1, max(x)+1, .01) #use more points for a smoother plot
y2 = numpy.polyval(coeffs, x2) #Evaluates the polynomial for each x2 value
plt.plot(x2, y2, label="deg=3")

在此处输入图像描述


注意:这是前面答案的一部分,如果您没有多变量数据,它仍然是相关的。而不是coeffs = mpf(...,使用coeffs = numpy.polyfit(x,y,3)

对于非多元数据集,最简单的方法可能是使用 numpy's polyfit

numpy.polyfit(x, y, deg, rcond=None, full=False, w=None, cov=False)

最小二乘多项式拟合。

p(x) = p[0] * x**deg + ... + p[deg]将一个多项式拟合deg到 points (x, y)。返回最小化平方误差的系数 p 向量。

于 2012-08-08T01:26:40.323 回答
0

稍微脱离上下文,因为结果函数不是多项式,但也许仍然很有趣。多项式拟合的一个主要问题是龙格现象:次数越高,发生的振荡就越剧烈。这也不仅仅是构建的,而且它会回来咬你。

作为补救措施,我不久前创建了Smoothfit。它解决了一个适当的最小二乘问题并给出了很好的结果,例如:

import numpy as np
import matplotlib.pyplot as plt
import smoothfit

x = [1, 4, 8, 11, 10, 15, 16]
y = [1, 3, 3, 4, 7, 11, 12]
a = 0.0
b = 17.0
plt.plot(x, y, 'kx')

lmbda = 3.0  # controls the smoothness
n = 100
u =  smoothfit.fit1d(x, y, a, b, n, lmbda)

x = np.linspace(a, b, n)
vals = [u(xx) for xx in x]
plt.plot(x, vals, "-")
plt.show()

在此处输入图像描述

于 2021-03-26T08:17:48.493 回答