checkpoints/training_summary.json

{
  "training_date": "2025-11-05 18:34:32",
  "config": {
    "batch_size": 16,
    "num_epochs": 20,
    "learning_rate": 2e-05,
    "device": "cuda"
  },
  "final_metrics": {
    "train_loss": 2.8002635020182116,
    "val_loss": 12.561226728844316,
    "train_acc": 0.9236858128678709,
    "val_acc": 0.7941429801894918
  },
  "num_discovered_risks": 7,
  "discovered_patterns": [
    "0",
    "1",
    "2",
    "3",
    "4",
    "5",
    "6"
  ]
}