使用 Keras/神经网络进行分类数据

Question

在 Keras 最小数据集的 CNN 示例中，他们告诉您如何建立一个良好的 CNN 网络来识别手写数字。问题是它没有告诉你如何预测新数字。

例如，如果我这样做，请给出图像：

model.predict(image)

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

Answer 1

您可以使用 numpy 的 argmax 找出概率最大的类

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