Downloads · 30 days
16
0% of all-time downloads
news-category-classification-distilbert
Yueh-Huan/news-category-classification-distilbert
news-category-classification-distilbert is a text classification model from Yueh-Huan. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as openrail.
This repository provides a distilbert model train on 210k news headlines from 2012 to 2022 from HuffPost. The model was trained by Yue
All-time downloads
45.3K
Public
Repo size
1.1 GB
Likes
20
Public
Repo makeup
Click a slice to open those files.
.h5268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- openrail
- Model type
- distilbert
- Access
- Public
- Created
- Mar 15, 2023
- Updated
- May 8, 2023
- SHA
- fed5542b
Try a prompt
Model card
From the Hugging Face model README
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/
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- openrail
- Languages
- en
- Created
- Mar 15, 2023
- Updated
- May 8, 2023