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

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)) 重塑了 X 和 y。然而它并没有奏效。

最佳答案

这对我来说很好。在重塑之前，请确保数组是 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])

关于python - sklearn : ValueError: Found input variables with inconsistent numbers of samples: [1, 6]，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/44181664/