Views
No views yet
| tag | meaning |
|---|---|
| PER | person name |
| LOC | location name |
| ORG | organization name |
| MISC | other name |
pip install flair)1from flair.data import Sentence
2from flair.models import SequenceTagger
3# load tagger
4tagger = SequenceTagger.load("helpmefindaname/mini-sequence-tagger-conll03")
5# make example sentence
6sentence = Sentence("George Washington went to Washington")
7# predict NER tags
8tagger.predict(sentence)
9# print sentence
10print(sentence)
11# print predicted NER spans
12print('The following NER tags are found:')
13# iterate over entities and print
14for entity in sentence.get_spans('ner'):
15 print(entity)Span [1,2]: "George Washington" [− Labels: PER (1.0)]
Span [5]: "Washington" [− Labels: LOC (1.0)]examples\ner\run_ner.py refers to this scriptpython examples\ner\run_ner.py --model_name_or_path hf-internal-testing/tiny-random-bert --dataset_name CONLL_03 --learning_rate 0.002 --mini_batch_chunk_size 1024 --batch_size 64 --num_epochs 100