Neural network 如何将softMax输出映射到MXNet中的标签
在深度学习中,预测通常使用一个热向量进行编码。我正在使用MXNet创建一个简单的神经网络,将动物图像分类为猫、狗、马等。当我调用MXNet的预测方法时,它会返回一个softmax输出。现在,我如何确定softmax输出中对应于最大概率的条目的索引是猫、狗或马。softmax输出仅提供一个数组,没有结果与相应标签的任何映射。这可能有助于回答您的问题。 本例使用预训练模型预测图像和synset数据集Neural network 如何将softMax输出映射到MXNet中的标签,neural-network,mxnet,Neural Network,Mxnet,在深度学习中,预测通常使用一个热向量进行编码。我正在使用MXNet创建一个简单的神经网络,将动物图像分类为猫、狗、马等。当我调用MXNet的预测方法时,它会返回一个softmax输出。现在,我如何确定softmax输出中对应于最大概率的条目的索引是猫、狗或马。softmax输出仅提供一个数组,没有结果与相应标签的任何映射。这可能有助于回答您的问题。 本例使用预训练模型预测图像和synset数据集