Downloads · 30 days
57
1% of all-time downloads
roberta-swahili
flax-community/roberta-swahili
roberta-swahili is a fill-mask model from flax-community. Use it when you need the model to fill a missing word. It is set up for transformers.
This model was trained using HuggingFace's Flax framework and is part of the JAX/Flax Community Week organized by HuggingFace. All training was done on a TPUv3-8 VM sponsored by the Google Cloud team.
All-time downloads
4.8K
Public
Repo size
29 GB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin422 MB · 47%
At a glance
- Task
- Fill-Mask
- Library
- transformers
- Model type
- roberta
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 25, 2021
- SHA
- 095dc7fd
Try a prompt
Datasets
Model card
From the Hugging Face model README
RoBERTa in Swahili
This model was trained using HuggingFace's Flax framework and is part of the JAX/Flax Community Week organized by HuggingFace. All training was done on a TPUv3-8 VM sponsored by the Google Cloud team.
How to use
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("flax-community/roberta-swahili")
model = AutoModelForMaskedLM.from_pretrained("flax-community/roberta-swahili")
print(round((model.num_parameters())/(1000*1000)),"Million Parameters")
105 Million Parameters
Training Data:
This model was trained on Swahili Safi
Results:
Eval metrics: {'f1': 86%}
This model was fine-tuned based off this model for the Zindi News Classification Challenge
More Details:
For more details and Demo please check HF Swahili Space
- Task
- Fill-Mask
- Library
- transformers
- Type
- roberta
- Languages
- sw
- Created
- Mar 2, 2022
- Updated
- Jul 25, 2021