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

ium 5

Browse Source
This commit is contained in:
Szymon Bartanowicz 2024-04-23 22:11:55 +02:00
parent bfc6feba86
commit f29fd743c3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model.py
View File

@ -26,7 +26,7 @@ preprocessor = ColumnTransformer(
pipeline = Pipeline(steps=[ pipeline = Pipeline(steps=[
('preprocessor', preprocessor), ('preprocessor', preprocessor),
('model', Sequential([ ('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(64, activation='relu'),
Dense(1) Dense(1)
])) ]))

2
predict.py
View File

@ -8,7 +8,7 @@ from sklearn.model_selection import train_test_split
loaded_model = tf.keras.models.load_model('powerlifting_model.h5') loaded_model = tf.keras.models.load_model('powerlifting_model.h5')
data = pd.read_csv('openpowerlifting.csv') 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']] features = data[['Sex', 'Age', 'BodyweightKg']]
target = data['TotalKg'] target = data['TotalKg']