如何将softMax输出映射到MXNet中的标签

Question

在深度学习中，预测通常使用一个热矢量进行编码。我正在使用MXNet创建一个简单的神经网络，将动物图像分类为猫，狗，马等。当我调用MXNet的Predict方法时，它返回softmax输出。现在，我如何确定softmax输出中与最大概率相对应的条目索引是Cats或Dogs或Horses。 softmax输出仅提供一个数组，而不会将结果与相应的标签进行任何映射。

Answer 1

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

此示例使用预训练模型来预测图像和synset数据集。