我正在测试基于cifar10 keras数据集的CNN模型。所以,基本上
prediction = mymodel.predict(x)
print(prediction)
打印出预测本身:
[[3.3675440e-04 5.7650192e-07 2.5850117e-02 5.4446888e-01 7.0444457e-02
1.7459875e-01 3.5874096e-03 1.8062484e-01 1.2066155e-06 8.6996079e-05]]
给定cifar10类标签,正确的语法是什么
classes = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
而不是打印预测的标签?
最佳答案
在您的情况下,应该是这样的:
prediction = mymodel.predict(x)
#return position of max
MaxPosition=np.argmax(prediction)
prediction_label=classes[MaxPosition]
print(prediction_label)
关于python - Keras:打印出预期的类(class)标签,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/59910151/