| { | |
| "base_model": "microsoft/deberta-v3-small", | |
| "mode": "binary", | |
| "epochs": 5, | |
| "learning_rate": 2e-05, | |
| "batch_size": 32, | |
| "gradient_accumulation_steps": 1, | |
| "warmup_ratio": 0.1, | |
| "weight_decay": 0.01, | |
| "max_length": 256, | |
| "label_smoothing": 0.15, | |
| "class_weights": [ | |
| 0.9379903674125671, | |
| 1.062009572982788 | |
| ], | |
| "seed": 42, | |
| "train_size": 3951, | |
| "val_size": 1909, | |
| "test_size": 2140, | |
| "train_label_dist": { | |
| "NOT_ACTION": 2098, | |
| "ACTION": 1853 | |
| }, | |
| "test_f1_macro": 0.8683538522408523, | |
| "test_f1_action": 0.853220696937698, | |
| "test_accuracy": 0.8700934579439252, | |
| "num_labels": 2, | |
| "label_map": { | |
| "NEUTRAL": 0, | |
| "POLICY": 0, | |
| "ACTION": 1 | |
| } | |
| } |