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)
