results.json

{
  "model": "efficientnet_b0",
  "dataset": "hf-vision/chest-xray-pneumonia",
  "seed": 42,
  "epochs": 5,
  "best_val_auc": 1.0,
  "test_metrics": {
    "loss": null,
    "accuracy": 0.8125,
    "precision": 0.7910447761194029,
    "recall": 0.9512820512820512,
    "f1": 0.8637951105937136,
    "roc_auc": 0.9037256191102344
  },
  "class_dist": {
    "train": [
      200,
      200
    ],
    "val": [
      8,
      8
    ],
    "test": [
      234,
      390
    ]
  }
}