bipin/malayalam-news-classifier

Malayalam news classifier

Overview

This model is trained on top of MalayalamBert for the task of classifying malayalam news headlines. Presently, the following news categories are supported:

• Business
• Sports
• Entertainment

Dataset

The dataset used for training this model can be found here.

Using the model with HF pipeline

from transformers import pipeline

news_headline = "ക്രിപ്‌റ്റോ ഇടപാടുകളുടെ വിവരങ്ങൾ ആവശ്യപ്പെട്ട് ആദായനികുതി വകുപ്പ് നോട്ടീസയച്ചു"
model = pipeline(task="text-classification", model="bipin/malayalam-news-classifier")

model(news_headline)
# Output
# [{'label': 'business', 'score': 0.9979357123374939}]

Contact

For feedback and questions, feel free to contact via twitter @bkrish_