distilbert-base-news-category-classification
This repository provides a distilbert model train on 210k news headlines from 2012 to 2022 from HuffPost. The model was trained by Yue
The training data can be found at the following Kaggle URL. https://www.kaggle.com/datasets/rmisra/news-category-dataset
Model and Project details: https://yueh-huan.com/posts/identify-news-category-based-on-news-headlines/
- Downloads last month
- 327
This model does not have enough activity to be deployed to Inference API (serverless) yet.
Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.