|
--- |
|
inference: False |
|
datasets: |
|
- arubenruben/portuguese_wikineural |
|
- Babelscape/wikineural |
|
language: |
|
- pt |
|
metrics: |
|
- f1 |
|
pipeline_tag: token-classification |
|
--- |
|
# Portuguese NER BERT-CRF Conll 2003 |
|
|
|
This model is a fine-tuned BERT model adapted for Named Entity Recognition (NER) tasks. It utilizes Conditional Random Fields (CRF) as the decoder. |
|
|
|
The model follows the Conll 2003 labeling scheme for NER. Additionally, it provides options for HAREM Default and Selective labeling schemes. |
|
|
|
## How to Use |
|
|
|
You can employ this model using the Transformers library's *pipeline* for NER, or incorporate it as a conventional Transformer in the HuggingFace ecosystem. |
|
|
|
```python |
|
from transformers import pipeline |
|
import torch |
|
import nltk |
|
|
|
ner_classifier = pipeline( |
|
"ner", |
|
model="arubenruben/NER-PT-BERT-CRF-Conll2003", |
|
device=torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu"), |
|
trust_remote_code=True |
|
) |
|
|
|
text = "FCPorto vence o Benfica por 5-0 no Estádio do Dragão" |
|
tokens = nltk.wordpunct_tokenize(text) |
|
result = ner_classifier(tokens) |
|
``` |
|
|
|
## Evaluation |
|
|
|
#### Testing Data |
|
|
|
The model was tested on the Portuguese Wikineural Dataset. |
|
|
|
### Results |
|
|
|
F1-Score: 0.951 |
|
|
|
## Citation |
|
|
|
Citation will be made available soon. |
|
|
|
**BibTeX:** |
|
:( |
