From c94695e20b63061619274e16357e7cbb79053e93 Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 14 May 2024 23:01:38 +0200 Subject: [PATCH] fix --- model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model.py b/model.py index d8bad31..d761a2d 100644 --- a/model.py +++ b/model.py @@ -36,7 +36,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:] -pipeline.fit(X_train_excluded, y_train, model__epochs=10, model__validation_split=0.1) +pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=10, model__validation_split=0.1) pipeline['model'].save('powerlifting_model.h5')