Update README.md
Browse files
README.md
CHANGED
|
@@ -29,12 +29,12 @@ inference:
|
|
| 29 |
|
| 30 |
# GoEmotions BERT Classifier
|
| 31 |
|
| 32 |
-
This is a fine-tuned **BERT-base-uncased** model for multi-label emotion classification on the [GoEmotions dataset](https://huggingface.co/datasets/
|
| 33 |
|
| 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
|
|
|
|
| 29 |
|
| 30 |
# GoEmotions BERT Classifier
|
| 31 |
|
| 32 |
+
This is a fine-tuned **BERT-base-uncased** model for multi-label emotion classification on the [GoEmotions dataset](https://huggingface.co/datasets/google-research-datasets/go_emotions), predicting 28 emotions (e.g., admiration, anger, joy, neutral).
|
| 33 |
|
| 34 |
## Model Details
|
| 35 |
|
| 36 |
- **Architecture**: BERT-base-uncased (110M parameters)
|
| 37 |
+
- **Training Data**: [GoEmotions](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
|