From a3da4f5c76cd089daeeea14bf6ba15666e6e7774 Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 14 May 2024 23:12:11 +0200 Subject: [PATCH] fix --- model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index 240a59a..86c8f82 100644 --- a/model.py +++ b/model.py @@ -35,8 +35,8 @@ pipeline = Pipeline(steps=[ pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae']) -X_train_excluded = X_train.iloc[1:, :] -y_train_excluded = y_train.iloc[1:, :] +X_train_excluded = X_train.iloc[1:] +y_train_excluded = y_train.iloc[1:] pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=10, model__validation_split=0.1)