ner english large EU AI Act Compliance Profile

English NER in Flair (large model)

This is the large 4-class NER model for English that ships with Flair.

F1-Score: 94,36 (corrected CoNLL-03)

Predicts 4 tags:

| tag | meaning | |---------------------------------|-----------| | PER | person name | | LOC | location name | | ORG | organization name | | MISC | other name |

Based on document-level XLM-R embeddings and FLERT.

Demo: How to use in Flair

Requires: Flair (pip install flair)

from flair.data import Sentence
from flair.models import SequenceTagger

# load tagger
tagger = SequenceTagger.load("flair/ner-english-large")

# make example sentence
sentence = Sentence("George Washington went to Washington")

# predict NER tags
tagger.predict(sentence)

# print sentence
print(sentence)

# print predicted NER spans
print('The following NER tags are found:')
# iterate over entities and print
for entity in sentence.get_spans('ner'):
    print(entity)

This yields the following output:

So, the entities "George Washington" (labeled as a person) and "Washington" (labeled as a location) are found in the sentence "George Washington went to Washington".

Training: Script to train this model

The following Flair script was used to train this model:

import torch

# 1. get the corpus
from flair.datasets import CONLL_03

corpus = CONLL_03()

# 2. what tag do we want to predict?
tag_type = 'ner'

# 3. make the tag dictionary from the corpus
tag_dictionary = corpus.make_tag_dictionary(tag_type=tag_type)

# 4. initialize fine-tuneable transformer embeddings WITH document context
from flair.embeddings import TransformerWordEmbeddings

embeddings = TransformerWord

Your obligations as a ner english large deployer

Track your progress

Sign up to track obligations, receive deadline reminders, and generate audit packages.

Tags

Languages

Library