Skip to content

Latest commit

 

History

History
50 lines (36 loc) · 1.69 KB

File metadata and controls

50 lines (36 loc) · 1.69 KB

Text_Emotion_Recognition

Text Emotion Recognition Web App

Training

Model Training Acc Training Loss Validation Acc Validation Loss Runtime
RoBERTa (base) 89.942 0.3164 74.538 1.24 12h 57m 50s
BERT (base) 38.449 1.789 39.002 1.854 10h 49min 33s
BERT (large) 44.676 1.624 43.67 1.794 11h 3m 24s
LSTM (64) 40.18 2.054 0.2847 2.229 2h 11m 41s
LSTM (128) 41.68 2.161 0.3018 2.30 2h 30m 50s
LSTM (256) 42.20 2.123 0.3044 2.267 3h 22m 4s
CNN (64 Filters) 39.09 1.881 0.2787 2.101 57m 49s
CNN (128 Filters) 38.98 1.860 0.2770 2.1 58m 34s
CNN (256 Filters) 44.84 1.728 0.307 2.046 59m 44s

Fine-Tuned Model: https://huggingface.co/junxtjx/roberta-base_TER/

RoBERTa_base: https://wandb.ai/wasabee/Text_Emotion_Recognition_RoBERTa?workspace=user-wasabee

BERT_large: https://wandb.ai/wasabee/Text_Emotion_Recognition_BERT_large_cased2?workspace=user-wasabee

BERT_base: https://wandb.ai/wasabee/Text_Emotion_Recognition_BERT_base_cased2?workspace=user-wasabee

LSTM_64_Cells: https://wandb.ai/wasabee/Text_Emotion_LSTM2_Cell_64?workspace=user-wasabee

LSTM_128_Cells: https://wandb.ai/wasabee/Text_Emotion_LSTM2_Cell_128?workspace=user-wasabee

LSTM_256_Cells: https://wandb.ai/wasabee/Text_Emotion_LSTM2_Cell_256?workspace=user-wasabee

CNN_64 https://wandb.ai/wasabee/Text_Emotion_CNN_64?workspace=user-wasabee

CNN_128 https://wandb.ai/wasabee/Text_Emotion_CNN_128?workspace=user-wasabee

CNN_256 https://wandb.ai/wasabee/Text_Emotion_CNN_256?workspace=user-wasabee

Dataset: https://data.world/crowdflower/sentiment-analysis-in-text