checkpoints/training_summary.json

{
  "training_date": "2025-11-06 19:51:32",
  "config": {
    "batch_size": 4,
    "num_epochs": 20,
    "learning_rate": 2e-05,
    "device": "cuda"
  },
  "final_metrics": {
    "train_loss": 3.522276586391842,
    "val_loss": 15.782539911401743,
    "train_acc": 0.9125228333671606,
    "val_acc": 0.7795004306632214
  },
  "num_discovered_risks": 7,
  "discovered_patterns": [
    "0",
    "1",
    "2",
    "3",
    "4",
    "5",
    "6"
  ]
}