Downloads · 30 days
40
1% of all-time downloads
roberta-swahili-news-classification
flax-community/roberta-swahili-news-classification
roberta-swahili-news-classification is a text classification model from flax-community. Use it when you need a label for a piece of text. 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.2K
Public
Parameters
105M
1.3 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin422 MB · 33%
Parameter types
How the weights are stored.
F32105M · 100%
Try a prompt
Datasets
Model card
From the Hugging Face model README
Swahili News Classification with RoBERTa
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.
This model was used as the base and fine-tuned for this task.
How to use
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("flax-community/roberta-swahili-news-classification")
model = AutoModelForSequenceClassification.from_pretrained("flax-community/roberta-swahili-news-classification")
Eval metrics: {'accuracy': 0.9153416415986249}
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Languages
- sw
- Created
- Mar 2, 2022
- Updated
- Mar 19, 2023