s464937/ium_464937
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

changed script

Browse Source
This commit is contained in:
Szymon Bartanowicz 2024-05-14 22:55:46 +02:00
parent 13369ee6ba
commit e8c6c3e7c3
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model.py
View File

@ -1,3 +1,4 @@
import sys
import pandas as pd import pandas as pd
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler, OneHotEncoder from sklearn.preprocessing import StandardScaler, OneHotEncoder
@ -7,7 +8,7 @@ from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense from tensorflow.keras.layers import Dense
import tensorflow as tf import tensorflow as tf
data = pd.read_csv('openpowerlifting.csv') data = pd.read_csv('./data/train.csv')
data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna() data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()
@ -34,6 +35,6 @@ pipeline = Pipeline(steps=[
pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae']) 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') pipeline['model'].save('powerlifting_model.h5')