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