optional code to customize markers with images

Author: SanderGi

notebooks/Eval/ASR_eval.ipynb

@@ -657,7 +657,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 65,
+   "execution_count": null,
    "id": "a11013dc",
    "metadata": {},
    "outputs": [
@@ -819,6 +819,10 @@
     "}\n",
     "SHOW_MAX_REGRESSION_ARROW = True\n",
     "\n",
+    "# optionally customize markers (e.g., with a flag) by providing a map from native-language to file path\n",
+    "from matplotlib.offsetbox import OffsetImage, AnnotationBbox\n",
+    "IMAGES: dict[str, str] = {}\n",
+    "\n",
     "for dataset, df_d in results.groupby('dataset'):      \n",
     "    # outliers\n",
     "    df_d = df_d[(df_d[\"cer\"] <= 1) & (df_d[\"wer\"] <= 1)] # drop outliers\n",
@@ -916,6 +920,17 @@
     "        (df_plot['wer_mean'] + df_plot['wer_std']) * 100, # type: ignore\n",
     "        color=wer_line.get_color(), alpha=0.2\n",
     "    )\n",
+    "    if top_lang in IMAGES:\n",
+    "        im = plt.imread(IMAGES[top_lang]) # type: ignore\n",
+    "        img = OffsetImage(im, zoom=15 / im.shape[0])  # tweak zoom to match marker size\n",
+    "        for x, y in zip(df_plot['release_date'], df_plot['wer_mean'] * 100):\n",
+    "            ab = AnnotationBbox(\n",
+    "                img,\n",
+    "                (x, y),\n",
+    "                frameon=False,\n",
+    "                box_alignment=(0.5, 0.5)\n",
+    "            )\n",
+    "            plt.gca().add_artist(ab)\n",
     "    if INCLUDE_CER:\n",
     "        cer_line, = plt.plot(df_plot['release_date'], df_plot['cer_mean'] * 100, '-s', color=CER_COLOR, label=f'CER ({top_lang})')\n",
     "        plt.fill_between(\n",