init

app.py

@@ -202,7 +202,9 @@ class DummyDataset(Dataset):
 
 def visualize(training_df, df, save_path, fast=False):
     if fast:
-        g_line = sns.lineplot(data=df, x='x', y='y', hue='Method', estimator=None, ci=None)
+        f = plt.figure(figsize=(3, 3))
+        g = f.add_subplot(111)
+        g_line = sns.lineplot(data=df, x='x', y='y', hue='Method', ax=g, estimator=None, ci=None)
         plt.xlim((Y_LB - B) / K, (Y_UB - B) / K)
         plt.ylim(Y_LB, Y_UB)
     else:
@@ -214,7 +216,7 @@ def visualize(training_df, df, save_path, fast=False):
                           height=8,
                           ratio=2,
                           estimator=None, ci=None,
-                          legend=False
+                          legend=False,
                           )
         g.ax_marg_x.remove()
         g_line = sns.lineplot(data=df, x='x', y='y', hue='Method', ax=g.ax_joint, estimator=None, ci=None)
@@ -224,7 +226,7 @@ def visualize(training_df, df, save_path, fast=False):
     plt.gca().axes.set_ylabel(r'$y$', fontsize=10)
 
     plt.savefig(save_path, bbox_inches='tight')
-    plt.clf()
+    plt.close()
 
 
 def clean_up_logs():