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
- 602