是否有可能获得 IOB 格式的预测？ - 内尔

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_)

产生

John B

Young I

goes O

for O

a O

walk O

. O

那么我认为您应该能够使用它来将预测转换为您需要的格式？

[前 2 条评论后更新]