Update README.md

README.md

@@ -42,6 +42,24 @@ Labels:
 27: neutral
 ```
 
+Training parameters:
+```yaml
+max_length: null
+batch_size: 32
+shuffle: True
+num_workers: 2
+pin_memory: False
+drop_last: False
+
+optimizer: adam
+lr: 0.00001
+weight_decay: 0
+
+problem_type: multi_label_classification
+
+num_epochs: 5
+```
+
 Evaluation results on test split of [lv_go_emotions](https://huggingface.co/datasets/SkyWater21/lv_go_emotions)
 |              |Precision|Recall|F1-Score|AUC-ROC|Support|
 |--------------|---------|------|--------|-------|-------|