File size: 3,660 Bytes
16d1326 4e1c286 33114ce 7652976 33114ce 06a342d 33114ce 16d1326 4e1c286 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
---
license: mit
language:
- en
pipeline_tag: text-classification
tags:
- code
- emotion
- emotions
widget:
- text: I'm so angry right now. I can't believe he did that to me.
example_title: anger
- text: I'm feeling disgusted by the smell of this food.
example_title: disgust
- text: I'm feeling very afraid of what might happen next.
example_title: fear
- text: I'm so joyful right now! This is the best day of my life.
example_title: joy
- text: >-
I'm feeling neutral about this situation. I don't really care one way or
another.
example_title: neutral
- text: I'm feeling really sad today after my dog passed away."
example_title: sadness
- text: I'm so surprised by what just happened! I never saw that coming.
example_title: surprise
- text: I'm feeling cheeky today. I'm going to play a little prank on my friend.
example_title: cheeky
- text: I'm feeling confused about what to do next. I need some guidance.
example_title: confuse
- text: I'm feeling curious about the world around me. There's so much to learn!
example_title: curious
- text: I'm feeling empathetic towards my friend who is going through a tough time.
example_title: empathetic
- text: I'm feeling grumpy today. Everything is annoying me!
example_title: grumpy
- text: I'm feeling guilty about what I did. I wish I could take it back.
example_title: guilty
- text: I'm feeling very energetic today. I'm ready to take on the world!
example_title: energetic
- text: I'm feeling impatient waiting for this movie to start.
example_title: impatient
- text: >-
I'm feeling so much love for my family right now. They mean everything to
me.
example_title: love
- text: I'm thinking about my future and what I want to achieve.
example_title: think
- text: >-
I'm feeling serious about this issue. It's important and needs to be
addressed.
example_title: serious
- text: >-
I'm feeling suspicious of what he's telling me. I think he's hiding
something.
example_title: suspicious
- text: I'm feeling whiny today. Everything is bothering me!
example_title: whiny
- text: I love football so much
example_title: love 2
- text: I'm reflecting on my experiences to gain insights
example_title: think 2
- text: >-
I borrowed money from a friend and haven't paid it back yet. Now I feel
ashamed.
example_title: guilty 2
- text: I'm starting to think that he's up to something.
example_title: suspicious 2
- text: We need to approach this matter with a sense of purpose
example_title: serious 2
---
# Emotion classification from 20 classes
## 20 Emotion labels
| id | label |
| --- | ---------- |
| 0 | anger |
| 1 | cheeky |
| 2 | confuse |
| 3 | curious |
| 4 | disgust |
| 5 | empathetic |
| 6 | energetic |
| 7 | fear |
| 8 | grumpy |
| 9 | guilty |
| 10 | impatient |
| 11 | joy |
| 12 | love |
| 13 | neutral |
| 14 | sadness |
| 15 | serious |
| 16 | surprise |
| 17 | suspicious |
| 18 | think |
| 19 | whiny |
## How to use
Here is how to use this model to get the emotion label of a given text:
```python
from transformers import AutoModelForSequenceClassification, pipeline
model_name = 'jitesh/emotion-english'
model = AutoModelForSequenceClassification.from_pretrained(model_name)
classifier = pipeline("text-classification", model=model, tokenizer=model_name)
text = "I can't wait any longer "
prediction = classifier(text)
print(prediction[0], text)
```
The above code outputs the following line.
```bash
{'label': 'impatient', 'score': 0.924211859703064} I can't wait any longer
``` |