线性回归期望X
是一个具有两个维度的数组,并且在内部需要X.shape[1]
来初始化一个np.ones
数组。因此,将X
转换为nx1数组就可以解决问题。所以,请替换:
regr.fit(x,y)
作者:
regr.fit(x[:,np.newaxis],y)
这将解决问题。演示:
>>> from sklearn import datasets
>>> from sklearn import linear_model
>>> clf = linear_model.LinearRegression()
>>> iris=datasets.load_iris()
>>> X=iris.data[:,3]
>>> Y=iris.target
>>> clf.fit(X,Y)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/lib/python2.7/dist-packages/sklearn/linear_model/base.py", line 363, in fit
X, y, self.fit_intercept, self.normalize, self.copy_X)
File "/usr/lib/python2.7/dist-packages/sklearn/linear_model/base.py", line 103, in center_data
X_std = np.ones(X.shape[1])
IndexError: tuple index out of range
>>> clf.fit(X[:,np.newaxis],Y)
LinearRegression(copy_X=True, fit_intercept=True, normalize=False)
使用以下代码绘制回归线:
>>> from matplotlib import pyplot as plt
>>> plt.scatter(X, Y, color='red')
<matplotlib.collections.PathCollection object at 0x7f76640e97d0>
>>> plt.plot(X, clf.predict(X[:,np.newaxis]), color='blue')
<matplotlib.lines.Line2D object at 0x7f7663f9eb90>
>>> plt.show()
![在此输入图片描述](https://istack.dev59.com/Vfs5z.webp)
linear_model
是什么?你是怎么得到它的? - Ffisegydd