Update README.md

This commit is contained in:
Hartmann 2021-06-16 09:32:20 +00:00 committed by huggingface-web
parent c1f87b8b1a
commit 75b9a09857
1 changed files with 16 additions and 2 deletions

View File

@ -1,7 +1,21 @@
Run emotion model with 3 lines of code on single text example using Hugging Face's pipeline command: ## Description
This deep neural network classifies the 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
## Application
a) Run emotion model with 3 lines of code on single text example using Hugging Face's pipeline command:
[![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) [![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)
Run emotion model on multiple examples and full datasets (e.g., .csv files): b) Run emotion model on multiple examples and full datasets (e.g., .csv files):
[![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) [![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)