metadata
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