Skip to content

Instantly share code, notes, and snippets.

@mohammedouahman
Created November 17, 2021 18:52
Show Gist options
  • Save mohammedouahman/16ecbb1ee5c3b99e363f840d3db9cbe5 to your computer and use it in GitHub Desktop.
Save mohammedouahman/16ecbb1ee5c3b99e363f840d3db9cbe5 to your computer and use it in GitHub Desktop.
predictions_single = probability_model.predict(img)
print(predictions_single)
[[8.8914348e-05 1.3264636e-13 9.9108773e-01 1.2658383e-10 8.1463791e-03
1.6905785e-08 6.7695131e-04 2.7492119e-17 5.1699739e-10 7.1339325e-17]]
plot_value_array(1, predictions_single[0], test_labels)
_ = plt.xticks(range(10), class_names, rotation=45)
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment