Downloads · 30 days
15
2% of all-time downloads
PuoBERTa-News
dsfsi/PuoBERTa-News
PuoBERTa-News is a text classification model from dsfsi. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as cc-by-4.0.
[](https://doi.org/10.5281/zenodo.8434795) [](https://arxiv.org/abs/2310.09141) 🤗 https://huggingface.co/dsfsi/PuoBERTa
All-time downloads
631
Public
Parameters
83.5M
668 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin334 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- cc-by-4.0
- Model type
- roberta
- Access
- Public
- Created
- Oct 13, 2023
- Updated
- Dec 4, 2023
- SHA
- 7661bf2e
Model card
From the Hugging Face model README
PuoBERTa-News: A Setswana Langauge Model Finetuned for News Categorisation
🤗 https://huggingface.co/dsfsi/PuoBERTa
Give Feedback 📑: DSFSI Resource Feedback Form{:target="_blank"}
A Roberta-based language model finetuned for News Categorisation.
Based on https://huggingface.co/dsfsi/PuoBERTa
Model Details
Model Description
This is a News Categorisation model for 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
News Categories
We use the IPTC news codes https://iptc.org/standards/newscodes/
- arts_culture_entertainment_and_media (Botsweretshi, setso, boitapoloso le bobegakgang)
- crime_law_and_justice (Bosenyi, molao le bosiamisi)
- disaster_accident_and_emergency_incident (Masetlapelo, kotsi le tiragalo ya maemo a tshoganyetso)
- economy_business_and_finance (Ikonomi, tsa kgwebo le tsa ditšhelete)
- education (Thuto)
- environment (Tikologo)
- health (Boitekanelo)
- politics (Dipolotiki)
- religion_and_belief (Bodumedi le tumelo)
- society (Setšhaba)
Training, Dev and Validation dataset https://huggingface.co/datasets/dsfsi/daily-news-dikgang.
Model Performance
Performance of models on Daily News Dikgang dataset
| Model | 5-fold Cross Validation F1 | Test F1 |
|---|---|---|
| Logistic Regression + TFIDF | 60.1 | 56.2 |
| NCHLT TSN RoBERTa | 64.7 | 60.3 |
| PuoBERTa | 63.8 | 62.9 |
| PuoBERTaJW300 | 66.2 | 65.4 |
Usage
Use this model for Part of text classification for Setswana.
Citation Information
Bibtex Reference
@inproceedings{marivate2023puoberta,
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},
year = {2023},
booktitle= {Artificial Intelligence Research. SACAIR 2023. Communications in Computer and Information Science},
url= {https://link.springer.com/chapter/10.1007/978-3-031-49002-6_17},
keywords = {NLP},
preprint_url = {https://arxiv.org/abs/2310.09141},
dataset_url = {https://github.com/dsfsi/PuoBERTa},
software_url = {https://huggingface.co/dsfsi/PuoBERTa}
}
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!
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- License
- cc-by-4.0
- Languages
- tn
- Created
- Oct 13, 2023
- Updated
- Dec 4, 2023