Changed dimensions of Heatmap.

Multiplicative Algorithm/Visualizations.py

@@ -66,7 +66,7 @@ def Heatmap(W1, W2):
     Ro = np.corrcoef(W1.T,W2.T)[len(W1.T):,:len(W2.T)]
     plt.imshow(Ro, cmap='hot', interpolation='nearest')
     plt.colorbar()
-    plt.title('Heatmap for correlation coefficients comparing two runs of W')
+    plt.title('Heatmap for comparing two runs of W')
     plt.xlabel('Columns of W1')
     plt.ylabel('Columns of W2')
     plt.savefig('Heatmap')
@@ -286,7 +286,7 @@ def permute_and_sort(W1, H1, W2, permute = False):
     if permute:
         permutation, sigs, SIGS = find_permutation(W1, W2)
     else:
-        plt.figure(figsize = (10,10), dpi=200)
+        plt.figure(figsize = (10,9), dpi=200)
         Heatmap(W1, W2)
         return W1, H1, W2, []