Final results

baseline_results.json

@@ -1,33 +1,56 @@
 {
     "experiment": "Baseline Original 3D U-Net",
-    "status": "training",
-    "last_completed_epoch": 50,
-    "next_epoch": 51,
+    "status": "completed",
+    "completed_epochs": 50,
     "best_epoch": 45,
     "best_validation_dice": 0.7721250271246268,
-    "config": {
-        "num_epochs": 50,
-        "batch_size": 1,
-        "num_classes": 4,
-        "model": "Original3DUNet",
-        "loss": "Dice + Cross Entropy",
-        "patch_size": [
-            64,
-            64,
-            64
-        ],
-        "modalities": [
-            "flair",
-            "t1",
-            "t1ce",
-            "t2"
-        ],
-        "train_patients": 294,
-        "validation_patients": 74,
-        "sampling": "Tumor-centered random sampling",
-        "dataset": "BraTS2020",
-        "excluded_patient": "BraTS20_Training_355",
-        "hf_repository": "Reduanul1997/RepositoryBaseLine3DuNet"
+    "best_epoch_metrics": {
+        "epoch": 45,
+        "learning_rate": 3.5111757055874303e-06,
+        "train_loss": 0.3746598991407018,
+        "val_loss": 0.476624794908472,
+        "train_dice": 0.84932605254679,
+        "val_dice": 0.7721250271246268,
+        "train_dice_macro_all": 0.8824622418971645,
+        "val_dice_macro_all": 0.8239281408112812,
+        "train_tumor_dice": 0.9395233504399889,
+        "val_tumor_dice": 0.9241284054564372,
+        "train_iou": 0.7384798154511153,
+        "val_iou": 0.6346612539007251,
+        "train_tumor_iou": 0.8859444013498974,
+        "val_tumor_iou": 0.8589578999420441,
+        "train_precision": 0.9468324232065657,
+        "val_precision": 0.9469938214247372,
+        "train_recall": 0.9323262580137129,
+        "val_recall": 0.9023411402697056,
+        "train_fnr": 0.06767374198628705,
+        "val_fnr": 0.09765885973029444,
+        "train_fpr": 0.015851784619549208,
+        "val_fpr": 0.014180152025839282,
+        "train_TP": 16700075,
+        "train_TN": 58220310,
+        "train_FP": 937761,
+        "train_FN": 1212190,
+        "val_TP": 3837129,
+        "val_TN": 14931465,
+        "val_FP": 214776,
+        "val_FN": 415286,
+        "val_dice_background": 0.9793374818712444,
+        "val_dice_ncr_net": 0.6698967303134118,
+        "val_dice_edema": 0.801320658082598,
+        "val_dice_enhancing": 0.8451576929778707,
+        "val_iou_background": 0.9595115569313988,
+        "val_iou_ncr_net": 0.503642646086615,
+        "val_iou_edema": 0.6685029349056847,
+        "val_iou_enhancing": 0.7318381807098756,
+        "epoch_seconds": 398.597907
+    },
+    "total_training_seconds": 14417.651626,
+    "files": {
+        "best_model": "/kaggle/working/baseline_3dunet_results/baseline_best_model.pth",
+        "latest_checkpoint": "/kaggle/working/baseline_3dunet_results/baseline_latest_checkpoint.pth",
+        "history_csv": "/kaggle/working/baseline_3dunet_results/baseline_history.csv",
+        "results_json": "/kaggle/working/baseline_3dunet_results/baseline_results.json"
     },
     "history": [
         {