{
  "model_name": "efficientnet_v2_s",
  "model_label": "EfficientNetV2-S",
  "class_names": [
    "Bacterial Red disease",
    "Bacterial diseases - Aeromoniasis",
    "Bacterial gill disease",
    "Fungal diseases Saprolegniasis",
    "Healthy Fish",
    "Parasitic diseases",
    "Viral diseases White tail disease"
  ],
  "checkpoint_path": "/workspace/artifacts/tuned_experiments/efficientnet_v2_s/best_model.pt",
  "device_used": "cuda",
  "accelerator_name": "NVIDIA GeForce RTX 3070 Ti Laptop GPU",
  "dataset_root": "/workspace/datasets/merged_balanced",
  "epochs_completed": 4,
  "training_seconds": 120.07,
  "best_epoch": 4,
  "best_validation_metrics": {
    "accuracy": 0.9778325123152709,
    "macro_precision": 0.9788300758439133,
    "macro_recall": 0.977832512315271,
    "macro_f1": 0.9778384028609978,
    "macro_auc_ovr": 0.999610724194553,
    "weighted_auc_ovr": 0.9996107241945529,
    "loss": 0.32660023244143704
  },
  "test_metrics": {
    "accuracy": 0.9827586206896551,
    "macro_precision": 0.9832862257395892,
    "macro_recall": 0.9827586206896551,
    "macro_f1": 0.9827068667021609,
    "macro_auc_ovr": 0.9995753354849668,
    "weighted_auc_ovr": 0.999575335484967,
    "loss": 0.3223129916073654
  }
}