Hugging Face's logo

wesleyacheng
/
news-topic-classification-with-bert

Text Classification
Transformers
PyTorch
Safetensors
English
distilbert
text-embeddings-inference
Model card Files
xet
 Community
1

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
7
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
Text Classification
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

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