|
--- |
|
|
|
widget: |
|
- text: "Oh wow. Where is that from??" |
|
- text: "This movie always makes me cry.." |
|
- text: "Really?! This is f++ked up!" |
|
|
|
--- |
|
|
|
## Description |
|
|
|
With this model, you can classify emotions in English text data. The model was trained on diverse datasets and predicts 7 emotions: |
|
|
|
1) anger |
|
2) disgust |
|
3) fear |
|
4) joy |
|
5) neutral |
|
6) sadness |
|
7) 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: |
|
|
|
[](https://colab.research.google.com/github/j-hartmann/emotion-english-distilroberta-base/blob/main/simple_emotion_pipeline.ipynb) |
|
|
|
b) Run emotion model on multiple examples and full datasets (e.g., .csv files) on Google Colab: |
|
|
|
[](https://colab.research.google.com/github/j-hartmann/emotion-english-distilroberta-base/blob/main/emotion_prediction_example.ipynb) |
|
|
|
## Contact |
|
|
|
Please reach out to jochen.hartmann@uni-hamburg.de if you have any questions or feedback. |
|
|
|
Thanks to S.D. and chrsiebert for their support in making this model available. |
