metadata

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

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), Moseli Mots'Oehli (@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 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.

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.

Email: [email protected]

Enjoy exploring Setswana through AI!