| { |
| "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 |
| } |
| } |
