在Keras中显示预测后的百分比

时间:2019-01-24 18:12:44

标签: python keras

我已经用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")

1 个答案:

答案 0 :(得分:3)

我对[0,1,...,9]类别做了一些测试。 函数predict返回一个数组,用于表示每个类别的概率。 这样一来,您只要打印max(prediction)即可,只要它是类别以及您需要的内容即可。