From e8c6c3e7c3e7ab3c2e150353a6530c5488f70a6e Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 14 May 2024 22:55:46 +0200 Subject: [PATCH] changed script --- model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index 74e81ad..9c57176 100644 --- a/model.py +++ b/model.py @@ -1,3 +1,4 @@ +import sys import pandas as pd from sklearn.model_selection import train_test_split from sklearn.preprocessing import StandardScaler, OneHotEncoder @@ -7,7 +8,7 @@ from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Dense import tensorflow as tf -data = pd.read_csv('openpowerlifting.csv') +data = pd.read_csv('./data/train.csv') data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna() @@ -34,6 +35,6 @@ pipeline = Pipeline(steps=[ pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae']) -pipeline.fit(X_train.iloc[:, 1:], y_train, model__epochs=10, model__validation_split=0.1) +pipeline.fit(X_train.iloc[:, 1:], y_train.iloc[:, 1:], model__epochs=int(sys.argv[1]), model__validation_split=0.1) pipeline['model'].save('powerlifting_model.h5')