使用Keras / Neural Networks进行分类数据

Question

在CNN example for the minst dataset for Keras中，他们会告诉您如何建立良好的CNN网络以识别手写数字。问题是它没有告诉你如何预测新的数字。

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

model.predict(image)

而不是告诉我它认为它是什么数字，而是给我一个10个数字的列表（可能是概率）

Answer 1

你可以使用numpy的argmax来找出具有最大概率的类

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