diff --git a/ium_zadanie6_evaluation.py b/ium_zadanie6_evaluation.py index 13df051..c1c590e 100644 --- a/ium_zadanie6_evaluation.py +++ b/ium_zadanie6_evaluation.py @@ -14,7 +14,7 @@ x_test = movies_test.copy() y_test = x_test.pop('rottentomatoes_audience_score') x_test.pop('Unnamed: 0') -model = keras.models.load_model('model_movies') +model = keras.models.load_model('model_movies.h5') y_predicted = model.predict(x_test, batch_size=64)