|
--- |
|
language: sw |
|
widget: |
|
- text: "Si kila mwenye makucha <mask> simba." |
|
datasets: |
|
- flax-community/swahili-safi |
|
--- |
|
|
|
## RoBERTa in Swahili |
|
|
|
This model was trained using HuggingFace's Flax framework and is part of the [JAX/Flax Community Week](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104) organized by [HuggingFace](https://huggingface.co). All training was done on a TPUv3-8 VM sponsored by the Google Cloud team. |
|
|
|
## How to use |
|
|
|
```python |
|
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](https://huggingface.co/datasets/flax-community/swahili-safi) |
|
|
|
#### **Results**: |
|
[MasakhaNER](https://github.com/masakhane-io/masakhane-ner) [](https://colab.research.google.com/drive/1OIurb4J91X7461NQXLCCGzjeEGJq_Tyl?usp=sharing) |
|
|
|
|
|
``` |
|
Eval metrics: {'f1': 86%} |
|
``` |
|
This [model](https://huggingface.co/flax-community/roberta-swahili-news-classification) was fine-tuned based off this model for the |
|
[Zindi News Classification Challenge](https://zindi.africa/hackathons/ai4d-swahili-news-classification-challenge) |
|
|
|
|
|
#### **More Details**: |
|
For more details and Demo please check [HF Swahili Space](https://huggingface.co/spaces/flax-community/Swahili) |
|
|
