简单线性回归模型

时间:2019-06-17 00:57:36

标签: python linear-regression

请帮助我。将数据拆分为

之后

npm install @jest/reporters

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

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

出现错误,指出数组必须是2D而不是1D数组。如何使其成为2D阵列。

1 个答案:

答案 0 :(得分: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)
相关问题