Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
# import load_model
from tensorflow.keras.models import load_model
# give the path to model directory to load the model
loaded_model = load_model('my_model/1/')
# predict function to predict the probabilities for each class 0-9
loaded_model.predict(test_images[0:1])
# predict_classes to get the class with highest probability
loaded_model.predict_classes(test_images[0:1])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.