Использование Keras/Neural Networks для категориальных данных

Question

В CNN example for the minst dataset for Keras они рассказывают вам, как сделать хорошую сеть CNN для распознавания ручных цифр. Проблема в том, что он не говорит вам, как прогнозировать новые цифры.

Например дает изображение, если я делаю это:

model.predict(image) 

вместо того, чтобы говорить мне, что цифры это думает, что это, вместо этого он дает мне список из 10 чисел (предположительно вероятности)

Answer 1

Вы можете использовать argmax numpy, чтобы узнать класс, который имеет максимальную вероятность

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