From a3b003c42210c4fdea058c5116495124f6d78955 Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 14 May 2024 23:19:35 +0200 Subject: [PATCH] fix --- model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index 86c8f82..ad48477 100644 --- a/model.py +++ b/model.py @@ -21,7 +21,8 @@ preprocessor = ColumnTransformer( transformers=[ ('num', StandardScaler(), ['Age', 'BodyweightKg']), ('cat', OneHotEncoder(), ['Sex']) - ] + ], + remainder='passthrough' ) pipeline = Pipeline(steps=[ @@ -38,6 +39,6 @@ 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_excluded, model__epochs=10, model__validation_split=0.1) +pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=int(sys.argv[1]), model__validation_split=0.1) pipeline['model'].save('powerlifting_model.h5')