是否可以以IOB格式获得预测？ -NER

Question

在评估我的NER模型时，我想将评估数据传递给预测方法，并获得IOB格式的预测输出。我想要这样做的原因是我需要使用seqeval来获得混淆矩阵，因为spaCy中没有这样的功能。这是否可能-产生与seqeval包兼容的输出？

Answer 1

您可以使用token.ent_iob访问IOB注释：

nlp = spacy.load('en_core_web_lg')
doc = nlp('John Young goes for a walk.')
for token in doc:
    print(token.text, token.ent_iob_)

产生

约翰B

年轻的我

去O

O

a O

走路O

。

那么我认为您应该可以使用它来将预测转换为所需的格式？

[在前2条评论后更新]