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

  • 本文关键字:-NER 格式 IOB 是否 spacy
  • 更新时间 :
  • 英文 :


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

您可以使用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

转到

对于O

a O

行走O

O

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

[在前2条评论后更新]

最新更新