🎨 Add y-axis legend

RasmussenLab · Mar 26, 2024 · af39e0a · af39e0a
1 parent 26ee826
commit af39e0a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/project/01_0_split_data.ipynb b/project/01_0_split_data.ipynb
@@ -1424,8 +1424,8 @@
     "fig, ax = plt.subplots(figsize=(6, 2))\n",
     "s = 1\n",
     "s_axes = pd.DataFrame({'medians': medians,\n",
-    "                       'validation split': splits.val_y.notna().sum(),\n",
-    "                       'training split': splits.train_X.notna().sum()}\n",
+    "                       'Validation split': splits.val_y.notna().sum(),\n",
+    "                       'Training split': splits.train_X.notna().sum()}\n",
     "                      ).plot.box(by='medians',\n",
     "                                 boxprops=dict(linewidth=s),\n",
     "                                 flierprops=dict(markersize=s),\n",
@@ -1434,7 +1434,7 @@
     "    _ = ax.set_xticklabels(ax.get_xticklabels(),\n",
     "                           rotation=45,\n",
     "                           horizontalalignment='right')\n",
-    "\n",
+    "    _ = ax.set_ylabel('Frequency')\n",
     "fname = args.out_figures / f'0_{group}_intensity_median_vs_prop_missing_boxplot_val_train'\n",
     "figures[fname.stem] = fname\n",
     "vaep.savefig(ax.get_figure(), fname)"

diff --git a/project/01_0_split_data.py b/project/01_0_split_data.py
@@ -904,8 +904,8 @@ def join_as_str(seq):
 fig, ax = plt.subplots(figsize=(6, 2))
 s = 1
 s_axes = pd.DataFrame({'medians': medians,
-                       'validation split': splits.val_y.notna().sum(),
-                       'training split': splits.train_X.notna().sum()}
+                       'Validation split': splits.val_y.notna().sum(),
+                       'Training split': splits.train_X.notna().sum()}
                       ).plot.box(by='medians',
                                  boxprops=dict(linewidth=s),
                                  flierprops=dict(markersize=s),
@@ -914,7 +914,7 @@ def join_as_str(seq):
     _ = ax.set_xticklabels(ax.get_xticklabels(),
                            rotation=45,
                            horizontalalignment='right')
-
+    _ = ax.set_ylabel('Frequency')
 fname = args.out_figures / f'0_{group}_intensity_median_vs_prop_missing_boxplot_val_train'
 figures[fname.stem] = fname
 vaep.savefig(ax.get_figure(), fname)