Dheeraj3103/distilbert-news-classifier-dheeraj

📚 DistilBERT News Classifier by Dheeraj

A text classification model fine-tuned on news headlines to classify them into one of four categories:

• 0 → Sports
• 1 → Business
• 2 → Tech
• 3 → 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)