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
- 654
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.