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

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

例如给出一张图像，如果我这样做:

model.predict(image)

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

最佳答案

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

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

关于python - 使用 Keras/神经网络处理分类数据，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/36039263/