checkpoints/training_summary.json

{
  "training_date": "2025-11-05 19:30:23",
  "config": {
    "batch_size": 4,
    "num_epochs": 20,
    "learning_rate": 2e-05,
    "device": "cuda"
  },
  "final_metrics": {
    "train_loss": 3.8652741019629158,
    "val_loss": 12.247559384727703,
    "train_acc": 0.9044043028211893,
    "val_acc": 0.7855297157622739
  },
  "num_discovered_risks": 7,
  "discovered_patterns": [
    "0",
    "1",
    "2",
    "3",
    "4",
    "5",
    "6"
  ]
}