Можно ли получить прогнозы в формате IOB? - NER

Question

При оценке моих моделей NER я хотел бы передать свои оценочные данные в метод прогнозирования и получить в качестве вывода прогнозы в формате IOB. Причина, по которой я этого хочу, заключается в том, что мне нужно использовать seqeval для получения матрицы путаницы, поскольку в spaCy такой возможности нет. Возможно ли это - произвести вывод, совместимый для использования с seqeval пакетом?

Answer 1

Вы можете получить доступ к аннотациям IOB с помощью token.ent_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_)

, что приводит к

Джону Б

Юному Я

идет O

для O

a O

ходьбы O

. O

Итак, я думаю, вы сможете использовать это для преобразования прогнозов в нужный вам формат?

[обновлено после первых 2 комментариев]