Downloads · 30 days
0
distilbert-news-classifier-dheeraj
Dheeraj3103/distilbert-news-classifier-dheeraj
distilbert-news-classifier-dheeraj is a text classification model from Dheeraj3103. 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.
A text classification model fine-tuned on news headlines to classify them into one of four categories:
Access
Public
Updated Jun 21, 2025
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
.json713 KB · 75%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Access
- Public
- Created
- Jun 21, 2025
- Updated
- Jun 21, 2025
- SHA
- 8dd9d89c
Try a prompt
Base models
Model card
From the Hugging Face model README
📚 DistilBERT News Classifier by Dheeraj
A text classification model fine-tuned on news headlines to classify them into one of four categories:
0 → Sports1 → Business2 → Tech3 → Science
🧠 Model Info
This model is built using DistilBERT and fine-tuned using Hugging Face's Trainer API.
It is trained on a small dataset of sample headlines, inspired by AG News, for demonstration and educational purposes.
🚀 Example Usage
from transformers import pipeline
classifier = pipeline("text-classification", model="Dheeraj3103/distilbert-news-classifier-dheeraj")
result = classifier("NASA finds evidence of water on Mars.")
print(result)
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Languages
- en
- Created
- Jun 21, 2025
- Updated
- Jun 21, 2025