Update README.md
Browse files
README.md
CHANGED
|
@@ -34,7 +34,7 @@ This is a fine-tuned **BERT-base-uncased** model for multi-label emotion classif
|
|
| 34 |
## Model Details
|
| 35 |
|
| 36 |
- **Architecture**: BERT-base-uncased (110M parameters)
|
| 37 |
-
- **Training Data**:
|
| 38 |
- **Loss Function**: Focal Loss (gamma=2)
|
| 39 |
- **Optimizer**: AdamW (lr=2e-5, weight_decay=0.01)
|
| 40 |
- **Epochs**: 5
|
|
|
|
| 34 |
## Model Details
|
| 35 |
|
| 36 |
- **Architecture**: BERT-base-uncased (110M parameters)
|
| 37 |
+
- **Training Data**: https://huggingface.co/datasets/google-research-datasets/go_emotions (58k Reddit comments, 28 emotions)
|
| 38 |
- **Loss Function**: Focal Loss (gamma=2)
|
| 39 |
- **Optimizer**: AdamW (lr=2e-5, weight_decay=0.01)
|
| 40 |
- **Epochs**: 5
|