Add label to colorbar in sns.heatmap (#815)

Co-authored-by: ArturoAmorQ <[email protected]>

Author: ArturoAmorQ

python_scripts/parameter_tuning_grid_search.py

@@ -240,7 +240,12 @@ def shorten_param(param_name):
 import seaborn as sns
 
 ax = sns.heatmap(
-    pivoted_cv_results, annot=True, cmap="YlGnBu", vmin=0.7, vmax=0.9
+    pivoted_cv_results,
+    annot=True,
+    cmap="YlGnBu",
+    vmin=0.7,
+    vmax=0.9,
+    cbar_kws={"label": "mean test accuracy"},
 )
 ax.invert_yaxis()