如何将SoftMax输出映射到MXNET中的标签

Question

在深度学习中，通常使用一个热量向量编码预测。我正在使用MXNET来创建一个简单的神经网络，该网络将动物的图像归类为猫，狗，马等。当我调用MXNET的预测方法时，它会使我返回SoftMax输出。现在，我如何确定与最大概率相对应的软马克斯输出中的条目索引是猫，狗或马。SoftMax输出仅给出一个阵列，而无需带有相应标签的结果映射。

Answer 1

这可能有助于回答您的问题。http://mxnet.io/tutorials/python/predict_imagenet.htmlhttps://github.com/dmlc/mxnet-notebooks/blob/master/python/how_to/predict.ipynb

此示例使用验证的模型来预测图像和同步数据集。