我正在测试基于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)# 返回最大值的位置:max_position = np.argmax(prediction)  prediction_label = classes[max_position]print(prediction_label)