News Topic Classifier
Model Description
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)
Results
| Metric | Score |
|---|---|
| Accuracy | 91.25% |
| F1-Score | 91.23% |
| Precision | 91.44% |
| Recall | 91.25% |
How to Use
from transformers import pipeline classifier = pipeline("text-classification", model="SAIRANaseem/news-classifier") result = classifier("Your news text here") print(result)
Classes
- LABEL_0: World
- LABEL_1: Sports
- LABEL_2: Business
- LABEL_3: Sci/Tech
- Downloads last month
- 9