update description emotion labels
Browse files
README.md
CHANGED
|
@@ -36,6 +36,17 @@ It may not generalize well to different domains or languages without further ada
|
|
| 36 |
|
| 37 |
The model was trained and evaluated on the "emotion" dataset, which includes labeled examples for emotion classification. The dataset consists of training, validation, and test sets, each containing text samples labeled with corresponding emotion categories.
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
## Training procedure
|
| 40 |
|
| 41 |
### Training hyperparameters
|
|
|
|
| 36 |
|
| 37 |
The model was trained and evaluated on the "emotion" dataset, which includes labeled examples for emotion classification. The dataset consists of training, validation, and test sets, each containing text samples labeled with corresponding emotion categories.
|
| 38 |
|
| 39 |
+
## Emotion Labels and Descriptions
|
| 40 |
+
|
| 41 |
+
The model predicts the following emotion labels:
|
| 42 |
+
|
| 43 |
+
- `0`: sadness
|
| 44 |
+
- `1`: joy
|
| 45 |
+
- `2`: love
|
| 46 |
+
- `3`: anger
|
| 47 |
+
- `4`: fear
|
| 48 |
+
- `5`: surprise
|
| 49 |
+
|
| 50 |
## Training procedure
|
| 51 |
|
| 52 |
### Training hyperparameters
|