---
license: openrail
language:
- en
---
# distilbert-base-news-category-classification
This repository provides a distilbert model train on 210k news headlines from 2012 to 2022 from [HuffPost](https://www.huffpost.com/). 
The model was trained by [Yue](https://github.com/yueeeeeee87)

The training data can be found at the following Kaggle URL.
[https://www.kaggle.com/datasets/rmisra/news-category-dataset](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/