Utilisation de Keras/Neural Networks pour les données catégorielles

Question

Dans le CNN example for the minst dataset for Keras, ils vous indiquent comment créer un bon réseau CNN pour reconnaître les chiffres écrits à la main. Le problème est qu'il ne vous dit pas comment prédire les nouveaux chiffres.

Par exemple donner une image, si je fais ceci:

model.predict(image) 

au lieu de me dire ce que les chiffres qu'il pense, il me donne au lieu une liste de 10 numéros (probablement probabilités)

Answer 1

Vous pouvez utiliser argmax de numpy pour trouver la classe qui a la probabilité maximale

import numpy as np 
probabilities = model.predict(image) 
classes = np.argmax(probabilities, axis=-1)