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