Changed Heatmap function to take Latex text for titles and save Heatmap fig as...

Changed Heatmap function to take Latex text for titles and save Heatmap fig as pdf trimmed of border

MultiplicativeAlgorithm/Visualizations.ipynb

@@ -1158,7 +1158,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.6.5"
   }
  },
  "nbformat": 4,

MultiplicativeAlgorithm/Visualizations.py

@@ -68,10 +68,11 @@ 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 comparing two runs of W')
-    plt.xlabel('Columns of W1')
-    plt.ylabel('Columns of W2')
-    plt.savefig('Heatmap')
+    plt.rc('text', usetex=True)
+    plt.title(r'Heatmap for comparing two runs of $W$')
+    plt.xlabel(r'Columns of $W_1$')
+    plt.ylabel(r'Columns of $W_2$')
+    plt.savefig('Heatmap.pdf', bbox_inches="tight")
     plt.show()
     
     return None