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

【问题标题】：Using Keras/ Neural Networks for categorical data使用 Keras/神经网络处理分类数据
【发布时间】：2016-03-16 14:56:27
【问题描述】：

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

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

model.predict(image)

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

【问题讨论】：

【解决方案1】：

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

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

【讨论】：