我已經(jīng)用 Keras 訓(xùn)練了模型，它根據(jù)帖子預(yù)測標(biāo)簽。訓(xùn)練后，它會循環(huán)超過 10 個測試帖子并預(yù)測標(biāo)簽。我還需要為每個帖子顯示預(yù)測的準(zhǔn)確性，但我不知道該怎么做...這是代碼示例。我將感謝您的幫助。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貢獻(xiàn)1818條經(jīng)驗 獲得超3個贊
我對某些類別進(jìn)行了一些測試，[0, 1, ..., 9]。該函數(shù)predict為每個類的概率返回一個數(shù)組。這樣，max(prediction)只要它是類別和您需要的任何內(nèi)容，您就可以打印。
添加回答
舉報
0/150
提交
取消