metadata
language: en
tags:
- sentiment
- emotion
- twitter
widget:
- text: Oh wow. I didn't know that.
- text: This movie always makes me cry..
Description
With this model, you can classify emotions in English text data. The model was trained on diverse datasets and predicts 7 emotions:
- anger
- disgust
- fear
- joy
- neutral
- sadness
- surprise
The model is a fine-tuned checkpoint of DistilRoBERTa-base.
Application
a) Run emotion model with 3 lines of code on single text example using Hugging Face's pipeline command on Google Colab:
b) Run emotion model on multiple examples and full datasets (e.g., .csv files) on Google Colab:
Contact
Please reach out to [email protected] if you have any questions or feedback.
Thanks to Samuel Domdey and chrsiebert for their support in making this model available.
Appendix
Please find an overview of the datasets used for training below:
| Name | anger | disgust | fear | joy | neutral | sadness | surprise |
|---|---|---|---|---|---|---|---|
| Crowdflower (2016) | Yes | No | No | Yes | Yes | Yes | Yes |
| Emotion Dataset, Elvis et al. (2018) | Yes | Yes | Yes | Yes | No | Yes | Yes |
| GoEmotions, Demszky et al. (2020) | Yes | Yes | Yes | Yes | Yes | Yes | Yes |
| ISEAR, Vikash (2018) | Yes | Yes | Yes | Yes | No | Yes | No |
| MELD, Poria et al. (2019) | Yes | Yes | Yes | Yes | Yes | Yes | Yes |
| SemEval-18 | Yes | No | Yes | Yes | No | Yes | No |