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
132
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.

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