我已经用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")
答案 0 :(得分:3)
我对[0,1,...,9]类别做了一些测试。
函数predict
返回一个数组,用于表示每个类别的概率。
这样一来,您只要打印max(prediction)
即可,只要它是类别以及您需要的内容即可。