在 FastText 文本分類輸出中打印分類報告

Question

我想使用 sklearn.metrics 庫打印分類報告和 F1 分數，但它需要預測的標簽。 Fasttext 僅給出 fig2 中的輸出，所以我想知道有沒有簡單的方法來獲取這些標簽並打印分類報告？

def train():
    self.model = fasttext.train_supervised(input='train.txt', wordNgrams=2, lr=1.0, epoch=10,
                                           bucket=200000, dim=300, loss='hs', pretrainedVectors='../cc.tr.300.vec')
    predict = self.model.test('test.txt')

    print(predict)

Fasttext 訓練和測試

輸出格式

Answer 1

不是將 test.txt 提供給 model.test，而是在循環中一一獲取所有句子並預測每個句子並插入到列表中。

self.pred = []
for sentence in self.sentences:
    self.pred.append(self.model.predict(sentence)[0][0].replace('__label__',''))

classification_report(y_true=self.real_tags, y_pred=self.pred, zero_division=0)