Archive final baseline evaluation artifacts

baseline_evaluation_results.json

@@ -0,0 +1,127 @@
+{
+    "experiment": "Baseline 3D U-Net",
+    "model": "Original3DUNet",
+    "best_model_epoch": 45,
+    "best_validation_dice_from_training": 0.772125,
+    "evaluation_split": "validation",
+    "num_validation_batches": 74,
+    "num_classes": 4,
+    "class_names": {
+        "0": "Background",
+        "1": "NCR/NET",
+        "2": "Edema",
+        "3": "Enhancing Tumor"
+    },
+    "foreground_macro_metrics": {
+        "Dice": 0.7578239404952702,
+        "IoU": 0.6187268333539934,
+        "Precision": 0.7816849800289972,
+        "Recall": 0.740964166304213,
+        "Sensitivity": 0.740964166304213
+    },
+    "tumor_metrics": {
+        "Dice": 0.9277145111848901,
+        "IoU": 0.865174919237252,
+        "Precision": 0.9423608690473698,
+        "Recall": 0.9135164588160428,
+        "Sensitivity": 0.9135164588160428,
+        "FNR": 0.08648354118395497,
+        "FPR": 0.017968357362018454,
+        "TP": 4312069,
+        "TN": 14414613,
+        "FP": 263746,
+        "FN": 408228
+    },
+    "per_class_metrics": {
+        "0": {
+            "class_name": "Background",
+            "TP": 14414613,
+            "TN": 4312069,
+            "FP": 408228,
+            "FN": 263746,
+            "Dice": 0.9772221468957192,
+            "IoU": 0.9554588456620434,
+            "Precision": 0.9724595305312923,
+            "Recall": 0.9820316426379809,
+            "Sensitivity": 0.9820316426379809,
+            "FNR": 0.017968357362018454,
+            "FPR": 0.08648354118395497
+        },
+        "1": {
+            "class_name": "NCR/NET",
+            "TP": 656767,
+            "TN": 17977341,
+            "FP": 252104,
+            "FN": 512444,
+            "Dice": 0.6320895903049031,
+            "IoU": 0.46208405596225705,
+            "Precision": 0.7226185014154844,
+            "Recall": 0.5617181158918231,
+            "Sensitivity": 0.5617181158918231,
+            "FNR": 0.4382818841081683,
+            "FPR": 0.013829493986240384
+        },
+        "2": {
+            "class_name": "Edema",
+            "TP": 1929680,
+            "TN": 16443872,
+            "FP": 568960,
+            "FN": 456144,
+            "Dice": 0.7901296846491226,
+            "IoU": 0.6530697336928837,
+            "Precision": 0.7722921269170399,
+            "Recall": 0.8088107085853743,
+            "Sensitivity": 0.8088107085853743,
+            "FNR": 0.1911892914146216,
+            "FPR": 0.03344299173706057
+        },
+        "3": {
+            "class_name": "Enhancing Tumor",
+            "TP": 993227,
+            "TN": 18058317,
+            "FP": 175077,
+            "FN": 172035,
+            "Dice": 0.8512525465317852,
+            "IoU": 0.7410267104068393,
+            "Precision": 0.8501443117544676,
+            "Recall": 0.8523636744354415,
+            "Sensitivity": 0.8523636744354415,
+            "FNR": 0.1476363255645499,
+            "FPR": 0.009601997302312443
+        }
+    },
+    "multiclass_confusion_matrix": [
+        [
+            14414613,
+            18849,
+            208375,
+            36522
+        ],
+        [
+            118371,
+            656767,
+            302510,
+            91563
+        ],
+        [
+            241100,
+            168052,
+            1929680,
+            46992
+        ],
+        [
+            48757,
+            65203,
+            58075,
+            993227
+        ]
+    ],
+    "tumor_binary_confusion_matrix": {
+        "TN": 14414613,
+        "FP": 263746,
+        "FN": 408228,
+        "TP": 4312069
+    },
+    "multiclass_confusion_matrix_image": "/kaggle/working/baseline_3dunet_results/baseline_multiclass_confusion_matrix.png",
+    "tumor_confusion_matrix_image": "/kaggle/working/baseline_3dunet_results/baseline_tumor_background_confusion_matrix.png"
+}