如何将 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

此示例使用预训练模型来预测图像和同义词集。