Skip to content

Instantly share code, notes, and snippets.

@lakshay-arora

lakshay-arora/predict-mnist-results.py

Created February 26, 2020 06:39
Show Gist options
  • Save lakshay-arora/d13790ce3ca072757a7510d7a740a21e to your computer and use it in GitHub Desktop.
Save lakshay-arora/d13790ce3ca072757a7510d7a740a21e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict-mnist-results.py
# 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