Named Entity Recognition

The following frameworks supported for performing NER in text:

DeepPavlov

• Source: https://github.com/nicolay-r/fast-ner
• List of the supported models.
• Limitation: Batch Size = 1

Example of the parameters for infer script:

--ner-framework "deeppavlov" \
--ner-model-name "ner_ontonotes_bert" \
--ner-types "ORG|PERSON|LOC|GPE" \

🤗 Transformers

• List of the supported models.
• Limitation: Batch Size = 1 due to the specifics of the AREkit API (see https://github.com/nicolay-r/AREkit/issues/535)

Example of the parameters for infer script:

--ner-framework="transformers" \
--ner-model-name "Babelscape/wikineural-multilingual-ner" \
--ner-types "ORG|PER|LOC|GPE" \