minhnguyen5293 commited on
Commit
0de8b4b
·
verified ·
1 Parent(s): f075b6a

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -0
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.