Big update

jameschapman19 · Oct 25, 2023 · 9703684 · 9703684
1 parent 2c7c9e2
commit 9703684
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/cca_zoo/visualisation/covariance.py b/cca_zoo/visualisation/covariance.py
@@ -85,14 +85,12 @@ def plot(self):
         sns.heatmap(
             self.train_covariances,
             annot=True,
-            cmap="coolwarm",
             ax=axs[0],
         )
         if self.test_covariances is not None:
             sns.heatmap(
                 self.test_covariances,
                 annot=True,
-                cmap="coolwarm",
                 ax=axs[1],
             )
         axs[0].set_title("Train Covariances")

diff --git a/cca_zoo/visualisation/scores.py b/cca_zoo/visualisation/scores.py
@@ -172,7 +172,7 @@ def _create_plot(self, x, y, hue, alpha=None, palette=None):
             ax,
         )
 
-    def plot(self, title=None):
+    def plot(self, title=""):
         dimensions = self.scores[0].shape[1]
         self.figures_ = []
 
@@ -226,7 +226,7 @@ def _create_plot(self, x, y, hue=None, palette=None):
 
 
 class SeparateScoreScatterDisplay(ScoreScatterDisplay):
-    def plot(self, title=None):
+    def plot(self, title=""):
         dimensions = self.scores[0].shape[1]
         self.train_figures_ = []
         self.test_figures_ = []

diff --git a/cca_zoo/visualisation/weights.py b/cca_zoo/visualisation/weights.py
@@ -46,7 +46,7 @@ def plot(self, **kwargs):
         self.weights_cov = [w.T @ w for w in self.weights]
         # loop through each view and have a heatmap of the covariance of the weights_
         for i, view_weights_cov in enumerate(self.weights_cov):
-            sns.heatmap(view_weights_cov, ax=axs[i], **self.kwargs)
+            sns.heatmap(view_weights_cov, ax=axs[i],annot=True, **self.kwargs)
             axs[i].set_title(self.view_labels[i])
         plt.tight_layout()
         self.figure_ = fig