checkpoints/training_summary.json

{
  "training_date": "2025-11-05 08:16:22",
  "config": {
    "batch_size": 16,
    "num_epochs": 20,
    "learning_rate": 2e-05,
    "device": "cuda"
  },
  "final_metrics": {
    "train_loss": 3.2153510323592593,
    "val_loss": 14.533901302781823,
    "train_acc": 0.9398213923279887,
    "val_acc": 0.7795004306632214
  },
  "num_discovered_risks": 7,
  "discovered_patterns": [
    "0",
    "1",
    "2",
    "3",
    "4",
    "5",
    "6"
  ]
}