Downloads · 30 days
9
9% of all-time downloads
SAIRANaseem/news-classifier
news-classifier is a text classification model from SAIRANaseem. Use it when you need a label for a piece of text. It is set up for transformers.
This model classifies news articles into 4 categories using DistilBERT.
Downloads · 30 days
9
9% of all-time downloads
All-time downloads
97
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
This model classifies news articles into 4 categories using DistilBERT.
Developed by: SAIRA NASEEM
Model type: Text Classification
Base Model: distilbert-base-uncased
Dataset: AG News (fancyzhx/ag_news)
| Metric | Score |
|---|---|
| Accuracy | 91.25% |
| F1-Score | 91.23% |
| Precision | 91.44% |
| Recall | 91.25% |
from transformers import pipeline classifier = pipeline("text-classification", model="SAIRANaseem/news-classifier") result = classifier("Your news text here") print(result)