在 Keras 中预测后显示百分比

Question

我已经用 Keras 训练了模型，它根据帖子预测标签。训练后，它会循环超过 10 个测试帖子并预测标签。我还需要为每个帖子显示预测的准确性，但我不知道该怎么做...这是代码示例。我将感谢您的帮助。

for i in range(10):

prediction = loaded_model.predict(np.array([x_test[i]]))

predicted_label = text_labels[np.argmax(prediction)]

print(test_posts.iloc[i][:50], "...")

print('Actual label:' + test_tags.iloc[i])

print("Predicted label: " + predicted_label + "\n")

Answer 1

我对某些类别进行了一些测试，[0, 1, ..., 9]。该函数predict为每个类的概率返回一个数组。这样，max(prediction)只要它是类别和您需要的任何内容，您就可以打印。