Visualize AUC results using bar plots

Author: AAdewunmi

notebooks/04_bias_analysis.ipynb

@@ -1164,6 +1164,33 @@
     "\n",
     "summary\n"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9c318890",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Visualize AUC results using bar plots\n",
+    "\n",
+    "import matplotlib.pyplot as plt\n",
+    "\n",
+    "def plot_slice_auc(title, auc_dict):\n",
+    "    keys = list(auc_dict.keys())\n",
+    "    values = list(auc_dict.values())\n",
+    "\n",
+    "    plt.figure(figsize=(6,4))\n",
+    "    plt.bar(keys, values, color=[\"#6c757d\",\"#0d6efd\",\"#198754\"])\n",
+    "    plt.ylabel(\"AUC\")\n",
+    "    plt.title(title)\n",
+    "    plt.ylim(0,1)\n",
+    "    plt.show()\n",
+    "\n",
+    "plot_slice_auc(\"AUC by Brightness Slice\", auc_brightness)\n",
+    "plot_slice_auc(\"AUC by Resolution Slice\", auc_resolution)\n",
+    "plot_slice_auc(\"AUC by Confidence Bucket\", auc_confidence)\n"
+   ]
   }
  ],
  "metadata": {