我正在尝试建立线性回归模型,但是由于某种原因我的模型无法建立。 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)
我在做错什么吗?