diff --git a/Multiplicative Algorithm/Images/Heatmap.png b/Multiplicative Algorithm/Images/Heatmap.png new file mode 100644 index 0000000000000000000000000000000000000000..7f11e943bef9c21968842d65ff6d2ea3a8f7e680 Binary files /dev/null and b/Multiplicative Algorithm/Images/Heatmap.png differ diff --git a/Multiplicative Algorithm/Visualizations.py b/Multiplicative Algorithm/Visualizations.py index 935f798b995aa85a4677056e27c25b774db2923a..cb86e9853444886df5955f9f40f50c38babd3194 100644 --- a/Multiplicative Algorithm/Visualizations.py +++ b/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, []