Spaces:
Sleeping
Sleeping
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,37 +1,37 @@
|
|
| 1 |
-
---
|
| 2 |
-
title: Emotion Intensity Prediction using Transformer Based Models
|
| 3 |
-
emoji: 🤩
|
| 4 |
-
colorFrom: purple
|
| 5 |
-
colorTo: indigo
|
| 6 |
-
sdk: streamlit
|
| 7 |
-
sdk_version: 1.
|
| 8 |
-
app_file: app.py
|
| 9 |
-
pinned: false
|
| 10 |
-
---
|
| 11 |
-
|
| 12 |
-
# Multitask Emotion Prediction Space
|
| 13 |
-
|
| 14 |
-
This Hugging Face Space hosts a deep learning model that predicts emotions and their intensities from text.
|
| 15 |
-
It utilizes a BERT-based architecture combined with lexicon features for enhanced performance.
|
| 16 |
-
|
| 17 |
-
**Features:**
|
| 18 |
-
- BERT-based text understanding.
|
| 19 |
-
- Integration of NRC VAD, NRC Emotion Lexicon, and NRC Hashtag Emotion Lexicon.
|
| 20 |
-
- Multi-task learning for emotion classification (joy, sadness, anger, fear) and intensity regression.
|
| 21 |
-
|
| 22 |
-
**How to use:**
|
| 23 |
-
Enter your text in the input box below and click "Predict Emotions" to see the model's output.
|
| 24 |
-
|
| 25 |
-
**Model Details:**
|
| 26 |
-
- Trained on dataset SemEval-2018 El-reg
|
| 27 |
-
- Uses `bert-base-uncased` from Hugging Face.
|
| 28 |
-
- `lex_dim`: 21 (number of combined lexicon features)
|
| 29 |
-
|
| 30 |
-
**Files included:**
|
| 31 |
-
- `app.py`: The Streamlit application code.
|
| 32 |
-
- `best_multitask_multilabel_model.pth`: Trained model weights.
|
| 33 |
-
- `*_scaler.pkl`: Joblib-saved feature scalers for lexicon features.
|
| 34 |
-
- `NRC-*.txt`: Lexicon data files.
|
| 35 |
-
|
| 36 |
-
---
|
| 37 |
Feel free to duplicate this Space and experiment!
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: Emotion Intensity Prediction using Transformer Based Models
|
| 3 |
+
emoji: 🤩
|
| 4 |
+
colorFrom: purple
|
| 5 |
+
colorTo: indigo
|
| 6 |
+
sdk: streamlit
|
| 7 |
+
sdk_version: 1.46.1
|
| 8 |
+
app_file: app.py
|
| 9 |
+
pinned: false
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Multitask Emotion Prediction Space
|
| 13 |
+
|
| 14 |
+
This Hugging Face Space hosts a deep learning model that predicts emotions and their intensities from text.
|
| 15 |
+
It utilizes a BERT-based architecture combined with lexicon features for enhanced performance.
|
| 16 |
+
|
| 17 |
+
**Features:**
|
| 18 |
+
- BERT-based text understanding.
|
| 19 |
+
- Integration of NRC VAD, NRC Emotion Lexicon, and NRC Hashtag Emotion Lexicon.
|
| 20 |
+
- Multi-task learning for emotion classification (joy, sadness, anger, fear) and intensity regression.
|
| 21 |
+
|
| 22 |
+
**How to use:**
|
| 23 |
+
Enter your text in the input box below and click "Predict Emotions" to see the model's output.
|
| 24 |
+
|
| 25 |
+
**Model Details:**
|
| 26 |
+
- Trained on dataset SemEval-2018 El-reg
|
| 27 |
+
- Uses `bert-base-uncased` from Hugging Face.
|
| 28 |
+
- `lex_dim`: 21 (number of combined lexicon features)
|
| 29 |
+
|
| 30 |
+
**Files included:**
|
| 31 |
+
- `app.py`: The Streamlit application code.
|
| 32 |
+
- `best_multitask_multilabel_model.pth`: Trained model weights.
|
| 33 |
+
- `*_scaler.pkl`: Joblib-saved feature scalers for lexicon features.
|
| 34 |
+
- `NRC-*.txt`: Lexicon data files.
|
| 35 |
+
|
| 36 |
+
---
|
| 37 |
Feel free to duplicate this Space and experiment!
|