使用 Keras/神经网络处理分类数据

Question

在 CNN example for the minst dataset for Keras 他们告诉你如何制作一个好的 CNN 网络来识别手写数字。问题是它没有告诉您如何预测新数字。

比如给一张图片，如果我这样做：

model.predict(image)

它没有告诉我它认为是什么数字，而是给了我一个包含 10 个数字（大概是概率）的列表

Answer 1

您可以使用 numpy 的 argmax 找出具有最大概率的class

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

Using Keras/ Neural Networks for categorical data