Update 'ium_zadanie6_evaluation.py'

This commit is contained in:
Zofia Galla 2021-05-15 19:22:55 +02:00
parent ad10e01844
commit d19358fe11

View File

@ -1,4 +1,4 @@
from tensorflow.keras.models import Sequential from tensorflow.keras.models import Sequential, load_model
from tensorflow.keras import layers from tensorflow.keras import layers
# from keras.layers import Flatten,Dense,Dropout, GlobalAveragePooling2D # from keras.layers import Flatten,Dense,Dropout, GlobalAveragePooling2D
from tensorflow.keras.optimizers import Adam from tensorflow.keras.optimizers import Adam
@ -13,7 +13,7 @@ x_test = movies_test.copy()
y_test = x_test.pop('rottentomatoes_audience_score') y_test = x_test.pop('rottentomatoes_audience_score')
x_test.pop('Unnamed: 0') x_test.pop('Unnamed: 0')
model = keras.models.load_model('model_movies.h5') model = load_model('model_movies.h5')
y_predicted = model.predict(x_test, batch_size=64) y_predicted = model.predict(x_test, batch_size=64)