Downloads · 30 days
878
11% of all-time downloads
bert-swahili-news-classification
flax-community/bert-swahili-news-classification
bert-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
8.1K
Public
Parameters
109M
1.3 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 33%
Parameter types
How the weights are stored.
F32109M · 100%
Try a prompt
Datasets
Model card
From the Hugging Face model README
Swahili News Classification with BERT
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/bert-swahili-news-classification")
model = AutoModelForSequenceClassification.from_pretrained("flax-community/bert-swahili-news-classification")
Eval metrics (10% valid set): {'accuracy': 0.9114740008594757}
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- sw
- Created
- Mar 2, 2022
- Updated
- Jul 15, 2023