pytorch roberta text-classification en transformers sentiment emotion twitter
.gitattributes README.md config.json merges.txt pytorch_model.bin special_tokens_map.json tokenizer.json tokenizer_config.json training_args.bin vocab.json
---
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:

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:

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](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:

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](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 Samuel Domdey and chrsiebert for their support in making this model available.