File size: 1,994 Bytes
a6b95ff 81ad031 61dfc25 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
---
license: cc-by-4.0
datasets:
- masakhane/masakhaner2
language:
- tn
metrics:
- f1
library_name: transformers
pipeline_tag: token-classification
---
# PuoBERTa-NER: A Setswana Langage Model Finetuned on MasakhaNER-2 for Named Entity Recognition.
A Roberta-based language model finetuned on MasakhaneNER-2 for Named Entity Recognition.
Based on [https://huggingface.co/dsfsi/PuoBERTa](https://huggingface.co/dsfsi/PuoBERTa)
## Model Details
### Model Description
This is a POS model trained on Setswana based on PuoBERTa and fineruned on MasakhaNER-2 Setswana.
- **Developed by:** Vukosi Marivate ([@vukosi](https://huggingface.co/@vukosi)), Moseli Mots'Oehli ([@MoseliMotsoehli](https://huggingface.co/@MoseliMotsoehli)) , Valencia Wagner, Richard Lastrucci and Isheanesu Dzingirai
- **Model type:** RoBERTa Model
- **Language(s) (NLP):** Setswana
- **License:** CC BY 4.0
### Model Performance
Performance of models on the [MasakhaNER-2](https://github.com/masakhane-io/masakhane-ner/tree/main/MasakhaNER2.0) downstream task.
| Model | Test Performance (f1 score) |
|---|---|
| **Multilingual Models** | |
| AfriBERTa | 83.2 |
| AfroXLMR-base | 87.7 |
| AfroXLMR-large | 89.4 |
| **Monolingual Models** | |
| NCHLT TSN RoBERTa | 74.2 |
| PuoBERTa | **78.2** |
| PuoBERTa+JW300 | **80.2** |
### Usage
Use this model for Part of Speech Tagging for Setswana.
```python
```
## Citation Information
Bibtex Refrence
```
@article{marivatePuoBERTa2023,
title={PuoBERTa: Training and evaluation of a curated language model for Setswana},
author={Vukosi Marivate and Moseli Mots'Oehli and Valencia Wagner and Richard Lastrucci and Isheanesu Dzingirai},
journal={ArXiv},
}
```
## Contributing
Your contributions are welcome! Feel free to improve the model.
## Model Card Authors
Vukosi Marivate
## Model Card Contact
For more details, reach out or check our [website](https://dsfsi.github.io/).
Email: [email protected]
**Enjoy exploring Setswana through AI!** |