Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/serializer.py Secret

Created May 20, 2021 16:07
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 aribornstein/a46a0e3e1c6fea67d54baf8cbb26dd44 to your computer and use it in GitHub Desktop.
Save aribornstein/a46a0e3e1c6fea67d54baf8cbb26dd44 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
serializer.py
model.serializer = Probabilities()
predictions = model.predict([
"data/hymenoptera_data/val/bees/65038344_52a45d090d.jpg",
"data/hymenoptera_data/val/bees/590318879_68cf112861.jpg",
"data/hymenoptera_data/val/ants/540543309_ddbb193ee5.jpg"
])
print(predictions)
# out: [[0.87, 0.13], [0.71, 0.29], [0.60, 0.40]]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment