Correlation analysis between image properties and model predictions

Author: AAdewunmi

notebooks/04_bias_analysis.ipynb

@@ -1191,6 +1191,26 @@
     "plot_slice_auc(\"AUC by Resolution Slice\", auc_resolution)\n",
     "plot_slice_auc(\"AUC by Confidence Bucket\", auc_confidence)\n"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b2d2386f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Correlation analysis between image properties and model predictions\n",
+    "\n",
+    "import seaborn as sns\n",
+    "import matplotlib.pyplot as plt\n",
+    "\n",
+    "corr = df[[\"brightness\",\"resolution\",\"prob_pneumonia\",\"prob_normal\"]].corr()\n",
+    "\n",
+    "plt.figure(figsize=(6,4))\n",
+    "sns.heatmap(corr, annot=True, cmap=\"Blues\", fmt=\".2f\")\n",
+    "plt.title(\"Feature Correlation Heatmap\")\n",
+    "plt.show()\n"
+   ]
   }
  ],
  "metadata": {