Skip to content

Instantly share code, notes, and snippets.

@satishgunjal
Created October 17, 2020 07:47
Show Gist options
  • Save satishgunjal/9d8dc946b3eae2b3ddab2f52948948f0 to your computer and use it in GitHub Desktop.
Save satishgunjal/9d8dc946b3eae2b3ddab2f52948948f0 to your computer and use it in GitHub Desktop.
predictions
probability_model = tf.keras.Sequential([model, tf.keras.layers.Softmax()])
predictions = probability_model.predict(test_images)
# 'predictions' will contain the prediction for each image in the training set. Lets check the first prediction
predictions[0]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment