CoolFace
Modelpublic

dev-isure/new-clasification-manish

sourceHugging Facemitupdated 4mo agoView on Hugging Face
0likes14downloads
Model Card

News Classifier — new-clasification-manish

Yeh model news headlines aur articles ko automatically 6 categories mein classify karta hai.

Developed by: Manish Negi — Isuremedia Base Model: distilbert-base-multilingual-cased Languages: English + Hindi Accuracy: 78% (Test Set)


Categories

IDCategoryExamples
0politicsElections, Government, Parliament, Modi, BJP
1sportsCricket, IPL, Football, Olympics
2businessSensex, RBI, Stock Market, Economy
3technologyiPhone, AI, Gadgets, Software
4entertainmentBollywood, Movies, Celebrities
5internationalWorld News, USA, China, War

How to Use

Install

bash
pip install transformers torch

Basic Usage

python
from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="dev-isure/new-clasification-manish"
)

# English news
result = classifier("Virat Kohli scores century in IPL match")
print(result)
# [{'label': 'sports', 'score': 0.95}]

# Hindi news
result = classifier("सेंसेक्स 500 अंक गिरा, बाजार में भारी बिकवाली")
print(result)
# [{'label': 'business', 'score': 0.91}]

Multiple Articles

python
news = [
    "PM Modi inaugurates new expressway in UP",
    "India beats Australia in T20 series",
    "RBI keeps repo rate unchanged at 6.5%",
    "Apple launches new iPhone 17 Pro",
    "Shah Rukh Khan's new film breaks box office records",
    "Russia Ukraine war ceasefire talks begin"
]

results = classifier(news)
for text, res in zip(news, results):
    print(f"{res['label']:<15} ({res['score']:.2f})  {text[:50]}")

Training Details

DetailValue
Base Modeldistilbert-base-multilingual-cased
Training Articles792
Validation Articles99
Test Articles100
Epochs3
Batch Size8
Learning Rate2e-5
Test Accuracy78%

Training Data Sources

  • —NDTV, Times of India, The Hindu, Hindustan Times
  • —Economic Times, ESPN Cricinfo
  • —BBC Hindi, Aaj Tak, Dainik Bhaskar
  • —Gadgets360, Bollywood Hungama, Pinkvilla

Limitations

  • —Health aur Crime categories is model mein nahi hain
  • —Hinglish (Hindi-English mixed) pe accuracy thodi kam ho sakti hai
  • —Zyada data se accuracy improve hogi