Downloads · 30 days
14
1% of all-time downloads
marcosgg/bert-base-pt-ner-enamex
bert-base-pt-ner-enamex is a token classification model from marcosgg. Use it when you need labels on individual words, such as names. It is set up for transformers. The card lists the license as agpl-3.0.
This is a NER model for Portuguese which uses the standard 'enamex' classes: LOC (geographical locations); PER (people); ORG (organizations); MISC (other entities).
Downloads · 30 days
14
1% of all-time downloads
All-time downloads
1.4K
Public
Repo size
867 MB
Likes
2
Public
Click a slice to open those files.
.bin433 MB · 100%
From the Hugging Face model README
This is a NER model for Portuguese which uses the standard 'enamex' classes: LOC (geographical locations); PER (people); ORG (organizations); MISC (other entities).
The model is based on BERTimbau Base, which has been fine-tuned using a combination of available corpora (see [1] for details).
There is an alternative model trained using BERTimbau Large: bert-large-pt-ner-enamex.
It was trained with a batch size of 8 and a learning rate of 2e-5 during 3 epochs. It achieved the following results on the test set (Precision/Recall/F1): 0.913/0.918/0.915.
[1] Pablo Gamallo, Marcos Garcia & Patricia Martín-Rodilla, 2019. NER and open information extraction for Portuguese notebook for IberLEF 2019 Portuguese named entity recognition and relation extraction tasks. In Proceedings of the Iberian Languages Evaluation Forum (IberLEF 2019) co-located with 35th Conference of the Spanish Society for Natural Language Processing (SEPLN 2019): 457-467.