From 35322b3bf90bfca2b8060efd7cd4a8a0f0ad0d3a Mon Sep 17 00:00:00 2001
From: Henry Webel <henry.webel@cpr.ku.dk>
Date: Thu, 16 Nov 2023 18:42:35 +0100
Subject: [PATCH] :art::construction: improve plots for Figure 2

(add more models)
- needs to be completed and cleaned-up
---
 project/03_2_best_models_comparison_fig2.py |  4 ++--
 vaep/plotting/defaults.py                   | 14 ++++++++++++--
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/project/03_2_best_models_comparison_fig2.py b/project/03_2_best_models_comparison_fig2.py
index f5e66baca..97c3e07e3 100644
--- 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
index 637f0815c..ca3d7851f 100644
--- 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):