简单线性回归模型

Question

请帮我。 将我的数据拆分为

X_train, y_train, X_test, y_test = train_test_split(X,y)

然后将它传递给我的线性回归模型即

linereg = LinearRegression().fit(X_train, y_train)

它带来了一个错误，说数组必须是二维而不是一维数组。 我怎样才能使它成为一个二维数组。

Answer 1

首先正确分割数据

X_train, x_test, Y_train,y_test=train_test_split(features,labels,train_size=0.7, test_size=0.3, random_state=2)

尝试使用重塑方法重塑 x_train 和 x_test。

x_test=x_test.reshape(-1,1)
x_train=x_train.reshape(-1,1)