🎨🚧 improve plots for Figure 2

(add more models) - needs to be completed and cleaned-up
RasmussenLab · Nov 16, 2023 · 35322b3 · 35322b3
1 parent 1569b97
commit 35322b3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/project/03_2_best_models_comparison_fig2.py b/project/03_2_best_models_comparison_fig2.py
@@ -90,7 +90,7 @@
 metrics = metrics.loc[ORDER_DATA, ORDER_MODELS]
 
 plt.rcParams['figure.figsize'] = [4.0, 2.0]
-matplotlib.rcParams.update({'font.size': 5})
+matplotlib.rcParams.update({'font.size': 6})
 
 ax = (metrics
       .plot
@@ -99,7 +99,7 @@
            ylabel=f"{METRIC} (log2 intensities)",
            color=COLORS_TO_USE_MAPPTING,
            width=.85,
-           fontsize=8
+           fontsize=7
            ))
 
 ax = vaep.plotting.add_height_to_barplot(ax, size=5)

diff --git a/vaep/plotting/defaults.py b/vaep/plotting/defaults.py
@@ -17,11 +17,21 @@
     'None': sns.color_palette()[7],
     'BPCA': sns.color_palette()[8],
     'MICE-CART': sns.color_palette()[9],
-    'SEQKNN': sns.color_palette()[6],
-    'MICE-NORM': sns.color_palette()[1],
+
 }
 # other_colors = sns.color_palette()[8:]
 other_colors = sns.color_palette("husl", 20)
+color_model_mapping['IMPSEQ'] = other_colors[0]
+color_model_mapping['IMPSEQROB'] = other_colors[1]
+color_model_mapping['MICE-NORM'] = other_colors[2]
+color_model_mapping['SEQKNN'] = other_colors[3]
+color_model_mapping['QRILC'] = other_colors[4]
+color_model_mapping['GSIMP'] = other_colors[5]
+color_model_mapping['MSIMPUTE'] = other_colors[6]
+color_model_mapping['MSIMPUTE_MNAR'] = other_colors[7]
+color_model_mapping['TRKNN'] = other_colors[8]
+color_model_mapping['SVDMETHOD'] = other_colors[9]
+other_colors = sns.color_palette()[10:]
 
 
 def assign_colors(models):