logasanjeev commited on
Commit
16e814d
·
verified ·
1 Parent(s): 0958065

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/goemotions), predicting 28 emotions (e.g., admiration, anger, joy, neutral).
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