This commit is contained in:
Jakub Zaręba 2023-05-10 18:58:18 +02:00
parent fea2fe4780
commit 9f1ca202e5

View File

@ -9,7 +9,7 @@ import os
model = tf.keras.models.load_model('model.h5') model = tf.keras.models.load_model('model.h5')
test_data = pd.read_csv('test_data.csv', sep=';') test_data = pd.read_csv('data.csv', sep=';')
test_data = pd.get_dummies(test_data, columns=['Sex', 'Medal']) test_data = pd.get_dummies(test_data, columns=['Sex', 'Medal'])
test_data = test_data.drop(columns=['Name', 'Team', 'NOC', 'Games', 'Year', 'Season', 'City', 'Sport', 'Event']) test_data = test_data.drop(columns=['Name', 'Team', 'NOC', 'Games', 'Year', 'Season', 'City', 'Sport', 'Event'])