File size: 705 Bytes
2237c3a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
---
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
|