MicahB commited on
Commit
92593b6
·
verified ·
1 Parent(s): 39e7df8

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +63 -0
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - distilroberta
5
+ - sentiment
6
+ - emotion
7
+ - twitter
8
+ - reddit
9
+ widget:
10
+ - text: Oh my God, he's lost it. He's totally lost it.
11
+ - text: What?
12
+ - text: Wow, congratulations! So excited for you!
13
+ library_name: transformers.js
14
+ ---
15
+
16
+ # Transformers.js conversion of [michellejieli/emotion_text_classifier](https://huggingface.co/michellejieli/emotion_text_classifier)
17
+
18
+ # Fine-tuned DistilRoBERTa-base for Emotion Classification 🤬🤢😀😐😭😲
19
+
20
+ # Model Description
21
+
22
+ DistilRoBERTa-base is a transformer model that performs sentiment analysis. I fine-tuned the model on transcripts from the Friends show with the goal of classifying emotions from text data, specifically dialogue from Netflix shows or movies. The model predicts 6 Ekman emotions and a neutral class. These emotions include anger, disgust, fear, joy, neutrality, sadness, and surprise.
23
+
24
+ The model is a fine-tuned version of [Emotion English DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/) and [DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base). This model was initially trained on the following table from [Emotion English DistilRoBERTa-base](https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/):
25
+
26
+ |Name|anger|disgust|fear|joy|neutral|sadness|surprise|
27
+ |---|---|---|---|---|---|---|---|
28
+ |Crowdflower (2016)|Yes|-|-|Yes|Yes|Yes|Yes|
29
+ |Emotion Dataset, Elvis et al. (2018)|Yes|-|Yes|Yes|-|Yes|Yes|
30
+ |GoEmotions, Demszky et al. (2020)|Yes|Yes|Yes|Yes|Yes|Yes|Yes|
31
+ |ISEAR, Vikash (2018)|Yes|Yes|Yes|Yes|-|Yes|-|
32
+ |MELD, Poria et al. (2019)|Yes|Yes|Yes|Yes|Yes|Yes|Yes|
33
+ |SemEval-2018, EI-reg, Mohammad et al. (2018) |Yes|-|Yes|Yes|-|Yes|-|
34
+
35
+ It was fine-tuned on:
36
+ |Name|anger|disgust|fear|joy|neutral|sadness|surprise|
37
+ |---|---|---|---|---|---|---|---|
38
+ |Emotion Lines (Friends)|Yes|Yes|Yes|Yes|Yes|Yes|Yes|
39
+
40
+ # How to Use
41
+
42
+ ```python
43
+ from transformers import pipeline
44
+ classifier = pipeline("sentiment-analysis", model="michellejieli/emotion_text_classifier")
45
+ classifier("I love this!")
46
+ ```
47
+
48
+ ```python
49
+ Output:
50
+ [{'label': 'joy', 'score': 0.9887555241584778}]
51
+ ```
52
+
53
+ # Contact
54
+
55
+ Please reach out to [michelleli1999@gmail.com](mailto:michelleli1999@gmail.com) if you have any questions or feedback.
56
+
57
+
58
+ # Reference
59
+
60
+ ```
61
+ Jochen Hartmann, "Emotion English DistilRoBERTa-base". https://huggingface.co/j-hartmann/emotion-english-distilroberta-base/, 2022.
62
+ Ashritha R Murthy and K M Anil Kumar 2021 IOP Conf. Ser.: Mater. Sci. Eng. 1110 012009
63
+ ```