Downloads · 30 days
9
13% of all-time downloads
agnews-distilbert-ft
starkdv123/agnews-distilbert-ft
agnews-distilbert-ft is a text classification model from starkdv123. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
This repository contains a DistilBERT model fine-tuned on the AG News dataset.
All-time downloads
67
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Sep 22, 2025
- Updated
- Sep 22, 2025
- SHA
- 43c3c970
Try a prompt
Datasets
Model card
From the Hugging Face model README
DistilBERT for AG News Classification — Full Fine-Tune
This repository contains a DistilBERT model fine-tuned on the AG News dataset.
Results
| Model | Test Accuracy | Macro F1 |
|---|---|---|
| Full Fine-Tune | 0.9426 | 0.9427 |
Confusion Matrix (Test)
World Sports Busines Sci/Tec
World 1797 13 42 48
Sports 10 1871 13 6
Busines 39 3 1692 166
Sci/Tec 26 8 62 1804
Usage
from transformers import pipeline
clf = pipeline("text-classification", model="starkdv123/agnews-distilbert-ft")
clf(["Markets tumble as central bank raises rates."])
Training (summary)
- Base:
distilbert-base-uncased - Epochs: 2, LR: 2e-5, batch: 16, max_len: 256, weight_decay: 0.01, fp16
Author
Karan D Vasa — https://huggingface.co/starkdv123
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Sep 22, 2025
- Updated
- Sep 22, 2025