README.md

metadata

license: apache-2.0
language:
  - en
metrics:
  - accuracy
base_model:
  - distilbert-base-uncased
pipeline_tag: text-classification
library_name: transformers
tags:
  - text-classification
  - news
  - distilbert
  - huggingface
  - ag_news

📚 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)