Edit model card

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.

Downloads last month
41
Safetensors
Model size
67M params
Tensor type
F32
·

Dataset used to train wesleyacheng/news-topic-classification-with-bert