| { | |
| "epoch": 4, | |
| "accuracy": 0.864, | |
| "precision_macro": 0.861, | |
| "recall_macro": 0.842, | |
| "f1_macro": 0.851, | |
| "loss": 0.428, | |
| "eval_samples": 20000, | |
| "confusion_matrix": { | |
| "simple": {"simple": 4800, "medium": 420, "complex": 130}, | |
| "medium": {"simple": 350, "medium": 5100, "complex": 520}, | |
| "complex": {"simple": 110, "medium": 440, "complex": 5150} | |
| }, | |
| "runtime_seconds": 1020, | |
| "gpu": "NVIDIA A100 40GB", | |
| "framework": "PyTorch", | |
| "transformers_version": "4.45.0" | |
| } | |