我已經用 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 回答
泛舟湖上清波郎朗
TA貢獻1818條經驗 獲得超3個贊
我對某些類別進行了一些測試，[0, 1, ..., 9]。該函數predict為每個類的概率返回一個數組。這樣，max(prediction)只要它是類別和您需要的任何內容，您就可以打印。
添加回答
舉報
0/150
提交
取消