Table of contents
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.
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.
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
- The Vietnamese VLSP model spells out the entire tag, though: PERSON, LOCATION, ORGANIZATION, MISCELLANEOUS.
- For packages with 18 named entity types, supported types include
CARDINAL(details can be found on page 21 of this OntoNotes documentation).
- The BSNLP19 dataset(s) use
- The Italian FBK dataset uses
- The Myanmar UCSY dataset uses
- The Japanese GSD dataset uses 22 tags:
- The Norwegian Norne dataset uses 8 tags for both NB and NN:
- The Persian Arman dataset uses 6 tags:
- The Turkish Starlang dataset uses 5 tags: