NER Models
Table of contents
System Performance on NER Corpora
In the table below you can find the performance of Stanza’s pretrained NER models. All numbers reported are micro-averaged F1 scores. We used canonical train/dev/test splits for all datasets except for the WikiNER datasets, for which we used random splits.
The Ukrainian model and its score was provided by a user.
Notes on NER Corpora
We have provided links to all NER datasets used to train the released models on our available NER models page. Here we provide notes on how to find several of these corpora:
Afrikaans: The Afrikaans data is part of the NCHLT corpus of South African languages. Van Huyssteen, G.B., Puttkammer, M.J., Trollip, E.B., Liversage, J.C., Eiselen, R. 2016. NCHLT Afrikaans Named Entity Annotated Corpus. 1.0.
Bulgarian: The Bulgarian BSNLP 2019 data is available from the shared task page. You can also find their dataset description paper.
Finnish: The Turku dataset used for Finnish NER training can be found on the Turku NLP website, and they also provide a Turku NER dataset description paper.
Hungarian: The dataset used for training our Hungarian NER system is a combination of 3 separate datasets: Business, Criminal, and NYTK. Two of these datasets can be found from this Szeged page, and the third can be found in this NYTK-NerKor github repo. A dataset description paper can also be found here.
Italian: The Italian FBK dataset was licensed to us from FBK. Paccosi T. and Palmero Aprosio A. KIND: an Italian Multi-Domain Dataset for Named Entity Recognition. LREC 2022
Myanmar: The Myanmar dataset is by special request from UCSY.
Swedish: The [SUC3 dataset] has two versions, one with the entries shuffled and another using the original ordering of the data. We make the shuffled version the default in order to expand the coverage of the model.
Vietnamese: The Vietnamese VLSP dataset is available by request from VLSP.
Tag category notes
- For packages with 4 named entity types, supported types include
PER
(Person),LOC
(Location),ORG
(Organization) andMISC
(Miscellaneous)- The Vietnamese VLSP model spells out the entire tag, though: PERSON, LOCATION, ORGANIZATION, MISCELLANEOUS.
- For packages with 18 named entity types, supported types include
PERSON
,NORP
(Nationalities/religious/political group),FAC
(Facility),ORG
(Organization),GPE
(Countries/cities/states),LOC
(Location),PRODUCT
,EVENT
,WORK_OF_ART
,LAW
,LANGUAGE
,DATE
,TIME
,PERCENT
,MONEY
,QUANTITY
,ORDINAL
andCARDINAL
(details can be found on page 21 of this OntoNotes documentation). - The BSNLP19 dataset(s) use
EVENT
,LOCATION
,ORGANIZATION
,PERSON
,PRODUCT
. - The Italian FBK dataset uses
LOC
,ORG
,PER
- The Myanmar UCSY dataset uses
LOC
(Location),NE
(Misc),ORG
(Organization),PNAME
(Person),RACE
,TIME
,NUM
- The Japanese GSD dataset uses 22 tags:
CARDINAL
,DATE
,EVENT
,FAC
,GPE
,LANGUAGE
,LAW
,LOC
,MONEY
,MOVEMENT
,NORP
,ORDINAL
,ORG
,PERCENT
,PERSON
,PET_NAME
,PHONE
,PRODUCT
,QUANTITY
,TIME
,TITLE_AFFIX
,WORK_OF_ART
- The Norwegian Norne dataset uses 8 tags for both NB and NN:
DRV
,EVT
,GPE
,LOC
,MISC
,ORG
,PER
,PROD
- The Persian Arman dataset uses 6 tags:
event
,fac
,loc
,org
,pers
,pro
- The Turkish Starlang dataset uses 5 tags:
LOCATION
,MONEY
,ORGANIZATION
,PERSON
,TIME