JuliusAlphonso's picture
Update README.md
6ff080b

Model description

This model was trained on the XED dataset and achieved
validation loss: 0.5995
validation acc: 84.28% (ROC-AUC)

Labels are based on Plutchik's model of emotions and may be combined: image

Framework versions

  • Transformers 4.6.1
  • Pytorch 1.8.1+cu101
  • Datasets 1.8.0
  • Tokenizers 0.10.3