NER позиция найденной сущности в тексте

Добрый день!

А правильно понимаю, что ner_model(['Curling World Championship will be held in Antananarivo']) возвращает только список сущностей и их типы, но не, например, позиции найденных сущностей в исходном тексте?

Если мне нужно получить эти самые позиции, я потом сам должен это сделать или есть какие-то встроенные функции?

Да, сама моделька работает с токенизированным текстом, поэтому позиции в исходном тексте теряются, встроенных решений в библиотеке для этого сейчас нет.
Можно попробовать либо восстановить из токенов, либо как-нибудь токенизировать самому перед подачей в модель с получением индексов.
ner_model([['Curling', 'World', 'Championship', 'will', 'be', 'held', 'in', 'Antananarivo']])

1 Like