satvik4577/mapnai-classifier

MAPNAI news classifier (Agent 2)

Multi-task news classifier used by Agent 2 of the MAPNAI pipeline. A fine-tuned distilroberta-base encoder with four heads:

Output	Values
domain	entertainment_movies, sports, other
category	Box Office, Reviews, Releases & Trailers, Casting & Production, Awards & Festivals, Celebrity News, Streaming/OTT · Cricket, Football, Tennis, Basketball, Motorsport, Olympics & Other Sports · Other
urgency_flag	true / false (threshold stored in label_config.json)
sentiment	-1.0 to 1.0

This is a custom architecture, so it is not loadable with AutoModelForSequenceClassification. Load it with agents/classifier_model.py from the MAPNAI repository:

hf download satvik4577/mapnai-classifier --local-dir models/classifier

from agents.classifier_model import NewsClassifierPredictor
p = NewsClassifierPredictor("models/classifier")
p.predict(["Kohli ruled out of the World Cup final with a hamstring injury"])

Training data: news from public RSS feeds plus the AG News, BBC News and HuffPost News Category datasets, labelled by Qwen2.5-7B-Instruct and validated against the taxonomy.

Version 1 notes: movie categories lean towards "Box Office"; retraining with class weighting and targeted data is planned.