File size: 1,577 Bytes
0de8b4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
base_model:
- google-bert/bert-base-uncased
---
# Model Card: **Emotion Detection Model for TV Show Dialogues**
## Model Overview
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:
- Happiness
- Sadness
- Anger
- Fear
- Surprise
- Disgust
- Love
- Excitement
- Anticipation
- Contentment
- Confusion
- Frustration
- Nostalgia
---
## Intended Use
- **Primary Use Case:** Emotion detection in textual data, particularly for analyzing media content like TV shows, movies, or social media.
- **Scope of Application:** Suitable for analyzing dialogue or conversational data to identify emotional trends, character development, or audience engagement.
---
## Dataset
### Source
- Dialogues from the TV show *Friends*.
### Labeling
- Dialogues were labeled with 2–3 emotions per text using OpenAI's ChatGPT API.
- Labels represent a rich variety of emotional states.
---
## Methodology
### Labeling Process
- ChatGPT API was used to classify text into multiple emotions.
- Outputs were reviewed for accuracy and adjusted to minimize misclassification.
### Model Architecture
- **Pre-trained Models Used:** BERT
- Fine-tuned on labeled dialogue data for emotion detection.
- Supports multi-label classification to capture multiple emotions simultaneously. |