Edit model card

First posted in my Kaggle.

Hello, I'm Wesley, nice to meet you! 👋

While I was making my Angry Birds Classifier to classify if tweets are angry or not, I thought why don't we add 2 more emotions! Joy and Sadness into the mix!

Here I created a Multiclass Text Classifier that classifies tweets as either having JOY, SADNESS, or ANGER.

I used the Twitter Emotion Dataset and BERT to do Transfer Learning with PyTorch and HuggingFace.

Downloads last month
18
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train wesleyacheng/twitter-emotion-classification-with-bert