diff --git a/evaluate.py b/evaluate.py index e9a75be..8026e26 100644 --- a/evaluate.py +++ b/evaluate.py @@ -9,7 +9,7 @@ import os 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 = test_data.drop(columns=['Name', 'Team', 'NOC', 'Games', 'Year', 'Season', 'City', 'Sport', 'Event'])