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