|
--- |
|
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 |
|
|
