Downloads · 30 days
11
0% of all-time downloads
wesleyacheng/news-topic-classification-with-bert
news-topic-classification-with-bert is a text classification model from wesleyacheng. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
BREAKING NEWS: We are making a news topic classifier!
Downloads · 30 days
11
0% of all-time downloads
All-time downloads
3.8K
Public
Parameters
67M
536 MB on disk
Likes
1
Public
Click a slice to open those files.
.bin268 MB · 50%
From the Hugging Face model README
BREAKING NEWS: We are making a news topic classifier!
First posted on my Kaggle.
Here I did transfer learning on BERT with the AG News Dataset to create a multi-class news topic classifier to classify news as either Business, Sports, Sci/Tech, or World.