python - sklearn : ValueError: Found input variables with inconsistent numbers of samples: [1, 6]

Question

X = [ 1994.  1995.  1996.  1997.  1998.  1999.]
y = [1.2 2.3 3.4 4.5 5.6 6.7]
clf = LinearRegression()
clf.fit(X,y)

这给出了上述错误。 X 和 y 都是 numpy 数组

如何消除此错误？

我试过给出的方法here并使用 X.reshape((-1,1)) 和 y.reshape((-1,1)) reshape X 和 y。然而，它并没有成功。

Answer 1

这对我来说很好用。在 reshape 之前，确保数组是 numpy 数组。

import numpy as np
from sklearn.linear_model import LinearRegression

X = np.asarray([ 1994.,  1995.,  1996.,  1997.,  1998.,  1999.])
y = np.asarray([1.2, 2.3, 3.4, 4.5, 5.6, 6.7])

clf = LinearRegression()
clf.fit(X.reshape(-1,1),y)


clf.predict([1997])
#Output: array([ 4.5])

clf.predict([2001])
#Output: array([ 8.9])