python-3.x - 发现输入变量样本数量不一致: [159, 40]

Question

我是机器学习新手。尝试线性回归并面临以下错误。请帮我解决它。这是我的代码:

  x=dataset.iloc[:,1:-1].values
y=dataset.iloc[:,-1].values
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
x_train,y_train,x_test,y_test=train_test_split(x,y,test_size=0.2)
regressor=LinearRegression()
regressor.fit(x_train,y_train)
y_pred=regressor.predict(x_test)

x.shape 为 [199,2]，y.shape 为 [199,]。执行代码后我收到以下错误:ValueError:发现输入变量的样本数量不一致:[159, 40]

Answer 1

train_test_split 的分割顺序看起来错误。应该是:

x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2)