无法为线性回归建立模型

时间:2020-10-29 20:50:27

标签: python

我正在尝试建立线性回归模型,但是由于某种原因我的模型无法建立。 X的形状为(10,2),Y的形状为(10,1)。经过训练的分割后,形状分别为(8,2),(2,2),(8,1)(2,1)。但是无论我尝试什么,模型都不会建立。

data = pd.read_csv("data.csv")


#clean data for ease of use
age_bed = data.iloc[0:10, 1:3]
age = data.iloc[0:10, 1:2]
bed = data.iloc[0:10, 2:3]
death = data.iloc[0:10, 3:4]

#define the x&y data
X = age_bed
Y = death

X_train, X_test, Y_train, Y_test = train_test_split(X,Y, test_size=0.2)
X_train.shape, Y_train.shape, X_test.shape, Y_test.shape

model = LinearRegression()
model.fit(X, Y)

仅输出:

Out[4]: LinearRegression()

假设返回类似以下内容的

LinearRegression(copy_X=True, fit_intercept=True, n_jobs=None,
         normalize=False)

我在做错什么吗?

0 个答案:

没有答案