【发布时间】:2012-08-05 01:44:18
【问题描述】:
如何在 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 页:
【问题讨论】:
-
即使在拟合多项式时,最小二乘回归仍然是线性的。只要方程是项的线性组合(例如多项式),相同的算法就可以工作。
-
你想为每个集合 X 生成一个公式,还是为所有集合生成一个公式?
标签: python matplotlib machine-learning regression scatter-plot