michellejieli commited on
Commit
df9a8b5
β€’
1 Parent(s): f515d4d

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +47 -0
README.md ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: "en"
3
+ tags:
4
+ - distilroberta
5
+ - sentiment
6
+ - emotion
7
+ - twitter
8
+ - reddit
9
+
10
+ widget:
11
+ - text: "Oh wow. I didn't know that."
12
+ - text: "This movie always makes me cry.."
13
+ - text: "Oh Happy Day"
14
+
15
+ ---
16
+
17
+ # Fine-tuned DistilRoBERTa-base for Emotion Classification
18
+
19
+ # Model Description
20
+
21
+ DistilRoBERTa-base is a transformers model, which is pre-trained on the Friends show to classify emotions from text. It predicts 6 Ekman emotions and a neutral class. These emotions include anger 🀬, disgust 🀒, fear 😨, joy πŸ˜€, neutral 😐, sadness 😭 and surprise 😲.
22
+
23
+ The model is a fine-tuned checkpoint of [DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base).
24
+
25
+ # How to Use
26
+
27
+ ```python
28
+ from transformers import pipeline
29
+ classifier = pipeline("sentiment-analysis", model="michellejieli/emotion_text_classifier")
30
+ classifier("I love this!")
31
+ ```
32
+
33
+ ```python
34
+ Output:
35
+ [{'label': 'joy', 'score': 0.9887555241584778}]
36
+ ```
37
+
38
+ # Contact
39
+
40
+ Please reach out to [michelle.li851@duke.edu](mailto:michelle.li851@duke.edu) if you have any questions or feedback.
41
+
42
+
43
+ # Reference
44
+
45
+ ```
46
+ Jochen Hartmann, "Emotion English DistilRoBERTa-base". https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/, 2022.
47
+ ```