Updating T-SNE Plot

Data_Plotting/Plot_TSNE.py

@@ -27,7 +27,8 @@ def plot_dimensionality_reduction(x, y, label_set, title):
     plt.title(title)
     if label_set[0].dtype == float:
         plt.scatter(x, y, c=label_set)
-        plt.colorbar()
+        cbar = plt.colorbar()
+        cbar.set_label('Average Density', fontsize=20)
         print("using scatter")
     else:
         for label in set(label_set):
@@ -35,6 +36,8 @@ def plot_dimensionality_reduction(x, y, label_set, title):
             plt.plot(x[cond], y[cond], marker='o', linestyle='none', label=label)
 
         plt.legend(numpoints=1)
+    plt.xlabel("Dimension 1")
+    plt.ylabel("Dimension 2")
 ########################################################################################################################
 """
 # Use for personal plotting