Nn-n commited on
Commit
92c010e
·
verified ·
1 Parent(s): 32ef4f0

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +98 -0
README.md ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - text-classification
5
+ - emotion-recognition
6
+ - sentiment-analysis
7
+ - MELD
8
+ - multimodal
9
+ pipeline_tag: text-classification
10
+ widget:
11
+ - text: "I'm so excited about this!"
12
+ - text: "That makes me really angry."
13
+ - text: "I'm feeling sad today."
14
+ license: mit
15
+ datasets:
16
+ - MELD
17
+ ---
18
+
19
+ # Emotion Classifier
20
+
21
+ This model classifies text into emotional categories based on the MELD (Multimodal EmotionLines Dataset) dataset. It can detect 7 emotions: anger, disgust, fear, joy, neutral, sadness, and surprise.
22
+
23
+ ## Model Details
24
+
25
+ - **Model Type:** Fine-tuned transformer-based text classification model
26
+ - **Base Model:** RoBERTa
27
+ - **Training Dataset:** MELD (Multimodal EmotionLines Dataset)
28
+ - **Number of Parameters:** ~125M
29
+ - **Sequence Length:** 128 tokens
30
+ - **Training Approach:** Fine-tuned with cross-validation
31
+
32
+ ## Intended Use
33
+
34
+ This model is designed to classify text into emotional categories. It can be used for:
35
+
36
+ - Sentiment analysis in customer feedback
37
+ - Emotion detection in conversations
38
+ - User experience research
39
+ - Content moderation
40
+ - Game development for adaptive emotional responses
41
+
42
+ ## Limitations
43
+
44
+ - The model was trained on scripted dialogues from TV shows, which may not fully represent natural conversations
45
+ - Short texts may be harder to classify accurately
46
+ - Cultural and contextual nuances might not be captured
47
+ - The model may reflect biases present in the training data
48
+
49
+ ## Performance
50
+
51
+ - **Accuracy:** [Insert your model's accuracy]
52
+ - **F1 Score:** [Insert your model's F1 score]
53
+ - **Training Dataset Size:** ~13,000 utterances
54
+
55
+ ## API Usage
56
+
57
+ ```python
58
+ import requests
59
+
60
+ API_URL = "https://api-inference.huggingface.co/models/YourUsername/emotion-classifier-meld"
61
+ headers = {"Authorization": "Bearer YOUR_HF_TOKEN"}
62
+
63
+ def query(payload):
64
+ response = requests.post(API_URL, headers=headers, json=payload)
65
+ return response.json()
66
+
67
+ output = query({"inputs": "I'm feeling excited!"})
68
+ print(output)
69
+ ```
70
+
71
+ ## Ethical Considerations
72
+
73
+ This model should be used responsibly. Consider the following ethical guidelines:
74
+
75
+ - Do not use this model to manipulate people's emotions
76
+ - Be transparent when using emotion detection in user-facing applications
77
+ - Do not make high-stakes decisions based solely on this model's outputs
78
+ - Consider privacy implications when analyzing personal communications
79
+
80
+ ## Citation
81
+
82
+ If you use this model, please cite the MELD dataset:
83
+
84
+ ```
85
+ @inproceedings{poria-etal-2019-meld,
86
+ title = "{MELD}: A Multimodal Multi-Party Dataset for Emotion Recognition in Conversations",
87
+ author = "Poria, Soujanya and
88
+ Hazarika, Devamanyu and
89
+ Majumder, Navonil and
90
+ Naik, Gautam and
91
+ Cambria, Erik and
92
+ Mihalcea, Rada",
93
+ booktitle = "Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics",
94
+ year = "2019",
95
+ publisher = "Association for Computational Linguistics",
96
+ pages = "527--536"
97
+ }
98
+ ```