minhnguyen5293
commited on
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model:
|
3 |
+
- google-bert/bert-base-uncased
|
4 |
+
---
|
5 |
+
# Model Card: **Emotion Detection Model for TV Show Dialogues**
|
6 |
+
|
7 |
+
## Model Overview
|
8 |
+
This model is trained to detect nuanced emotions in text data, specifically focusing on dialogues from the TV show *Friends* and additional curated online content. By leveraging advanced deep learning architectures such as BERT and GPT, the model performs multi-label classification to identify up to 2–3 emotions per input dialogue. The targeted emotions include:
|
9 |
+
- Happiness
|
10 |
+
- Sadness
|
11 |
+
- Anger
|
12 |
+
- Fear
|
13 |
+
- Surprise
|
14 |
+
- Disgust
|
15 |
+
- Love
|
16 |
+
- Excitement
|
17 |
+
- Anticipation
|
18 |
+
- Contentment
|
19 |
+
- Confusion
|
20 |
+
- Frustration
|
21 |
+
- Nostalgia
|
22 |
+
|
23 |
+
---
|
24 |
+
|
25 |
+
## Intended Use
|
26 |
+
- **Primary Use Case:** Emotion detection in textual data, particularly for analyzing media content like TV shows, movies, or social media.
|
27 |
+
- **Scope of Application:** Suitable for analyzing dialogue or conversational data to identify emotional trends, character development, or audience engagement.
|
28 |
+
|
29 |
+
---
|
30 |
+
|
31 |
+
## Dataset
|
32 |
+
### Source
|
33 |
+
- Dialogues from the TV show *Friends*.
|
34 |
+
|
35 |
+
### Labeling
|
36 |
+
- Dialogues were labeled with 2–3 emotions per text using OpenAI's ChatGPT API.
|
37 |
+
- Labels represent a rich variety of emotional states.
|
38 |
+
|
39 |
+
---
|
40 |
+
## Methodology
|
41 |
+
### Labeling Process
|
42 |
+
- ChatGPT API was used to classify text into multiple emotions.
|
43 |
+
- Outputs were reviewed for accuracy and adjusted to minimize misclassification.
|
44 |
+
|
45 |
+
### Model Architecture
|
46 |
+
- **Pre-trained Models Used:** BERT
|
47 |
+
- Fine-tuned on labeled dialogue data for emotion detection.
|
48 |
+
- Supports multi-label classification to capture multiple emotions simultaneously.
|