From f29fd743c3fb7467fa2b31dd3b76f65ba30921b3 Mon Sep 17 00:00:00 2001
From: Szymon Bartanowicz <szymon@bartanowicz.pl>
Date: Tue, 23 Apr 2024 22:11:55 +0200
Subject: [PATCH] ium 5

---
 model.py   | 2 +-
 predict.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/model.py b/model.py
index 796f46f..dad53bd 100644
--- a/model.py
+++ b/model.py
@@ -26,7 +26,7 @@ preprocessor = ColumnTransformer(
 pipeline = Pipeline(steps=[
     ('preprocessor', preprocessor),
     ('model', Sequential([
-        Dense(64, activation='relu', input_dim=4),  # Liczba wejść musi zgadzać się z wynikowym wymiarem preprocessingu
+        Dense(64, activation='relu', input_dim=4),
         Dense(64, activation='relu'),
         Dense(1)
     ]))
diff --git a/predict.py b/predict.py
index 270e7a7..1273567 100644
--- a/predict.py
+++ b/predict.py
@@ -8,7 +8,7 @@ from sklearn.model_selection import train_test_split
 loaded_model = tf.keras.models.load_model('powerlifting_model.h5')
 
 data = pd.read_csv('openpowerlifting.csv')
-data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()  # Usunięcie wierszy z brakującymi danymi
+data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()
 features = data[['Sex', 'Age', 'BodyweightKg']]
 target = data['TotalKg']