dkp2701/BERT-based-Multiclass-Emotion-Classification

BERT-based-Emotion-Classification

• BERT-based Emotion Classification.
• A custom Pytorch model which uses BERT for multiclass emotion classification.
• There are 8 different classes. Labels: anxiety, bipolar, depression, fear, grateful, happy, sad, stress
• Version 1: Achieved an accuracy of 88%

Model Huggingface Repo

• BERT-based-Emotion-Classification-v1: https://huggingface.co/dkp2701/BERT-based-Multiclass-Emotion-Classification

Dataset

• https://www.kaggle.com/datasets/deepakprajapati07/emotion-detection/data