added test acc
Browse files
README.md
CHANGED
|
@@ -16,7 +16,9 @@ This model fine-tunes DistilBERT for a multi-class emotion classification task.
|
|
| 16 |
## Training and Evaluation
|
| 17 |
- Training Dataset: dair-ai/emotion (16,000 examples)
|
| 18 |
- Validation Dataset: dair-ai/emotion (2,000 examples)
|
|
|
|
| 19 |
- Validation Accuracy: 96.6 %
|
|
|
|
| 20 |
- Training Time: 18,078.18s (5h 1m 18s)
|
| 21 |
|
| 22 |
## Hyperparameters
|
|
|
|
| 16 |
## Training and Evaluation
|
| 17 |
- Training Dataset: dair-ai/emotion (16,000 examples)
|
| 18 |
- Validation Dataset: dair-ai/emotion (2,000 examples)
|
| 19 |
+
- Test Dataset: dair-ai/emotion (2,000 examples)
|
| 20 |
- Validation Accuracy: 96.6 %
|
| 21 |
+
- Test Accuracy: 93.2 %
|
| 22 |
- Training Time: 18,078.18s (5h 1m 18s)
|
| 23 |
|
| 24 |
## Hyperparameters
|