Skip to content

Instantly share code, notes, and snippets.

@lakshay-arora
Created February 26, 2020 06:39
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
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