Upload modality_classifier_metadata.json with huggingface_hub

modality_classifier_metadata.json

@@ -0,0 +1,62 @@
+{
+  "config": {
+    "data_dir": null,
+    "output_dir": "/content/training_runs/modality_classifier",
+    "hf_token": "***redacted***",
+    "hf_dataset_repo": "iraqigold/Medical-Modality-Dataset",
+    "hf_model_repo": "iraqigold/Medical-Modality-Classifier",
+    "download_hf": false,
+    "upload_to_hf": true,
+    "backbone": "resnet18",
+    "epochs": 15,
+    "batch_size": 32,
+    "num_workers": 0,
+    "lr": 0.0001,
+    "weight_decay": 0.0001,
+    "val_ratio": 0.15,
+    "test_ratio": 0.1,
+    "seed": 42,
+    "save_every": 3,
+    "upload_every_epochs": 3,
+    "resume_training": true,
+    "force_balance": true
+  },
+  "classes": [
+    "CT",
+    "MRI",
+    "US",
+    "XRAY"
+  ],
+  "class_mapping": {
+    "CT": 0,
+    "MRI": 1,
+    "US": 2,
+    "XRAY": 3
+  },
+  "train_samples": 6000,
+  "val_samples": 1200,
+  "test_samples": 800,
+  "best_val_accuracy": 0.9991666666666666,
+  "test_metrics": {
+    "test_loss": 0.0007467886476661079,
+    "test_accuracy": 1.0,
+    "per_class_accuracy": {
+      "CT": 1.0,
+      "MRI": 1.0,
+      "US": 1.0,
+      "XRAY": 1.0
+    },
+    "per_class_correct": {
+      "CT": 206,
+      "MRI": 201,
+      "US": 178,
+      "XRAY": 215
+    },
+    "per_class_total": {
+      "CT": 206,
+      "MRI": 201,
+      "US": 178,
+      "XRAY": 215
+    }
+  }
+}