clean tsne module

Author: mingukkang

src/main.py

@@ -95,10 +95,10 @@ def main():
             "StudioGAN dose not support the evalutation protocol that uses the test dataset on imagenet, tiny imagenet, and custom datasets"
 
     if train_config['distributed_data_parallel']:
-        msg = "StudioGAN does not support image visualization, k_nearest_neighbor, interpolation, and frequency_analysis with DDP. " +\
+        msg = "StudioGAN does not support image visualization, k_nearest_neighbor, interpolation, frequency, and tsne analysis with DDP. " +\
             "Please change DDP with a single GPU training or DataParallel instead."
-        assert train_config['image_visualization'] + train_config['k_nearest_neighbor'] + \
-            train_config['interpolation'] + train_config['frequency_analysis'] + train_config['tsne_analysis'] == 0, msg
+        assert train_config['image_visualization'] + train_config['k_nearest_neighbor'] + train_config['interpolation'] +\
+            train_config['frequency_analysis'] + train_config['tsne_analysis'] == 0, msg
 
     hdf5_path_train = make_hdf5(model_config['data_processing'], train_config, mode="train") \
         if train_config['load_all_data_in_memory'] else None

src/utils/misc.py

@@ -418,15 +418,19 @@ def plot_tsne_scatter_plot(df, tsne_results, flag, run_name, logger):
 
     df['tsne-2d-one'] = tsne_results[:,0]
     df['tsne-2d-two'] = tsne_results[:,1]
+    # x="tsne-2d-one", y="tsne-2d-two",
     plt.figure(figsize=(16,10))
     sns.scatterplot(
         x="tsne-2d-one", y="tsne-2d-two",
         hue="labels",
         palette=sns.color_palette("hls", 10),
         data=df,
         legend="full",
-        alpha=0.3
-    )
+        alpha=0.5
+    ).legend(fontsize = 15, loc ='upper right')
+    plt.title("TSNE result of {flag} images".format(flag=flag), fontsize=25)
+    plt.xlabel('', fontsize=7)
+    plt.ylabel('', fontsize=7)
     plt.savefig(save_path)
     logger.info("Save image to {}".format(save_path))