| { |
| "model_name": "roberta-base", |
| "training_config": { |
| "max_length": 128, |
| "batch_size": 16, |
| "grad_accum_steps": 4, |
| "effective_batch_size": 64, |
| "learning_rate": 2e-05, |
| "weight_decay": 0.01, |
| "warmup_ratio": 0.1, |
| "label_smoothing": 0.0, |
| "epochs_trained": 5, |
| "max_epochs": 5, |
| "early_stopping_patience": 2, |
| "seed": 42 |
| }, |
| "test_metrics": { |
| "loss": 0.1989, |
| "accuracy": 0.9215, |
| "f1": 0.9197, |
| "precision": 0.9431, |
| "recall": 0.8975 |
| }, |
| "training_log": [ |
| { |
| "epoch": 1.0, |
| "eval_loss": 0.21930797398090363, |
| "eval_accuracy": 0.9154668860551214, |
| "eval_f1": 0.9150650960942344, |
| "eval_precision": 0.9275240888144114, |
| "eval_recall": 0.9029363784665579 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_loss": 0.21582643687725067, |
| "eval_accuracy": 0.9164952694364459, |
| "eval_f1": 0.9156626506024096, |
| "eval_precision": 0.9331075359864521, |
| "eval_recall": 0.8988580750407831 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_loss": 0.22042229771614075, |
| "eval_accuracy": 0.9127930892636775, |
| "eval_f1": 0.9140308191403081, |
| "eval_precision": 0.9088709677419354, |
| "eval_recall": 0.9192495921696574 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_loss": 0.2514384686946869, |
| "eval_accuracy": 0.9127930892636775, |
| "eval_f1": 0.9135752140236445, |
| "eval_precision": 0.91320293398533, |
| "eval_recall": 0.9139477977161501 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_loss": 0.1989288628101349, |
| "eval_accuracy": 0.9214638157894737, |
| "eval_f1": 0.919714165615805, |
| "eval_precision": 0.9431034482758621, |
| "eval_recall": 0.8974569319114027 |
| } |
| ], |
| "confusion_matrix": [ |
| [ |
| 1147, |
| 66 |
| ], |
| [ |
| 125, |
| 1094 |
| ] |
| ], |
| "training_time_minutes": 15.3, |
| "timestamp": "2026-03-26T11:49:50.790207", |
| "data_sizes": { |
| "train": 41332, |
| "validation": 4862, |
| "test": 2432 |
| } |
| } |
