dkp2701
/

BERT-based-Multiclass-Emotion-Classification

Text Classification

Text_based_emotion_classification

Model card Files Files and versions Community

Edit model card

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

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Examples

Text Classification

Unable to determine this model's library. Check the docs .

Model tree for dkp2701/BERT-based-Multiclass-Emotion-Classification

Base model

google-bert/bert-base-uncased

Finetuned

(2174)

this model