Skip to content

Instantly share code, notes, and snippets.

@lakshay-arora

lakshay-arora/display_results.py

Created February 26, 2020 07:21
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save lakshay-arora/d60df332c78bc4b4dc689b0770acd0a9 to your computer and use it in GitHub Desktop.
Save lakshay-arora/d60df332c78bc4b4dc689b0770acd0a9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
display_results.py
# get the prediction array
predictions = predictions['predictions']
# print the actual image and the predicted result
for i, prediction in enumerate(predictions):
print("Prediction: ",np.argmax(prediction))
show(i,test_images[i])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment