CoolFace
Modelpublic

Keshav0308/multilingual-topic-classifier

sourceHugging Facemitupdated 5mo agoView on Hugging Face
0likes9downloads
Model Card

🌍 Multilingual Topic Classifier

A multilingual text classification model fine-tuned on the SIB-200 dataset, capable of classifying text into 7 topics across 205 languages.

Model Details

  • β€”Base model: xlm-roberta-base
  • β€”Task: Text Classification (Topic)
  • β€”Languages: 205
  • β€”Developed by: Keshav0308

Topics

LabelDescription
🌍 geographyGeographic content
πŸ”¬ science/technologyScience and tech content
🎬 entertainmentEntertainment content
πŸ›οΈ politicsPolitical content
πŸ₯ healthHealth and medical content
✈️ travelTravel content
⚽ sportsSports content

Performance

MetricScore
Test Accuracy69.17%
Test F1 Macro67.62%
Languages205

Usage

python
from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="Keshav0308/multilingual-topic-classifier"
)

# Works in any language!
classifier("The patient was diagnosed with pneumonia.")
# {'label': 'health', 'score': 0.999}

classifier("El equipo ganΓ³ el campeonato mundial de fΓΊtbol.")
# {'label': 'sports', 'score': 0.999}

Training Data

Fine-tuned on SIB-200 β€” a massively multilingual dataset with 205 languages.

  • β€”Train samples: 143,705
  • β€”Validation samples: 20,295
  • β€”Test samples: 41,820