| --- |
| license: mit |
| language: |
| - en |
| base_model: |
| - google-bert/bert-base-uncased |
| pipeline_tag: text-classification |
| tags: |
| - Text_based_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 |
|
|
