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

fix

Browse Source
This commit is contained in:
Szymon Bartanowicz 2024-05-14 23:25:25 +02:00
parent a3b003c422
commit a860f43d29
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

@ -8,7 +8,9 @@ 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('./data/train.csv') # data = pd.read_csv('./data/train.csv')
data = pd.read_csv('./openpowerlifting.csv')
print(data)
data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna() data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()
@ -22,7 +24,6 @@ preprocessor = ColumnTransformer(
('num', StandardScaler(), ['Age', 'BodyweightKg']), ('num', StandardScaler(), ['Age', 'BodyweightKg']),
('cat', OneHotEncoder(), ['Sex']) ('cat', OneHotEncoder(), ['Sex'])
], ],
remainder='passthrough'
) )
pipeline = Pipeline(steps=[ pipeline = Pipeline(steps=[