| { | |
| "model_name": "bert-base-uncased", | |
| "max_length": 128, | |
| "num_labels": 31, | |
| "batch_size": 16, | |
| "learning_rate": 2e-5, | |
| "num_epochs": 5, | |
| "warmup_ratio": 0.1, | |
| "weight_decay": 0.01, | |
| "max_grad_norm": 1.0, | |
| "lr_scheduler_type": "linear", | |
| "output_dir": "Classification/checkpoints", | |
| "logging_steps": 10, | |
| "evaluation_strategy": "epoch", | |
| "save_strategy": "epoch", | |
| "load_best_model_at_end": true, | |
| "metric_for_best_model": "f1", | |
| "seed": 42 | |
| } |