File size: 1,098 Bytes
1e23579 bde0da2 1ffaf36 1e23579 bde0da2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | # Hyperparameter Search Results for `BRlkl/BingoGuard-bert-base-pt-optimized`
This model was selected through a randomized search over a hyperparameter space.
The best performing model was chosen based on the highest average F1 score across multiple benchmark datasets.
## Best Hyperparameters
```
learning_rate: 5e-05
per_device_train_batch_size: 32
num_train_epochs: 8
weight_decay: 0.05
lr_scheduler_type: cosine
```
## Final Benchmark Results (Full Dataset Training)
| dataset | accuracy | f1_score | recall | precision |
|:----------------------------------|-----------:|-----------:|---------:|------------:|
| BRlkl/BingoGuard-train-test-pt | 0.897773 | 0.946133 | 0.897773 | 1 |
| BRlkl/openai-moderation-eval-pt | 0.69881 | 0.651994 | 0.908046 | 0.508584 |
| BRlkl/WildGuardTest-pt | 0.839317 | 0.810021 | 0.771883 | 0.852123 |
| BRlkl/XSTest-pt | 0.831111 | 0.824885 | 0.895 | 0.764957 |
| BRlkl/toxic-chat-pt (40% holdout) | 0.97296 | 0.836795 | 0.927632 | 0.762162 | |