jiawei011 commited on
Commit
a50507e
1 Parent(s): 6e9ac12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -227,7 +227,7 @@ def visualize(training_df, df, save_path, fast=False, epoch=None):
227
  plt.gca().axes.set_xlabel(r'$x$')
228
  plt.gca().axes.set_ylabel(r'$y$')
229
 
230
- plt.savefig(save_path, bbox_inches='tight', dpi=50 if fast else 100)
231
  plt.close()
232
 
233
 
@@ -256,7 +256,7 @@ def run(num1, num2, num3, num4, num5, random_seed, mode):
256
  regress(train_loader, training_df)
257
  make_video()
258
  if mode == 0:
259
- text = "Press \"Start Regressing\" if your are happy with the training data. Regression takes ~10s."
260
  else:
261
  text = "Press \"Prepare Training Data\" before moving the sliders. You may also change the random seed."
262
  training_data_plot = 'training_data.png' if mode == 0 else None
 
227
  plt.gca().axes.set_xlabel(r'$x$')
228
  plt.gca().axes.set_ylabel(r'$y$')
229
 
230
+ plt.savefig(save_path, bbox_inches='tight', dpi=200)
231
  plt.close()
232
 
233
 
 
256
  regress(train_loader, training_df)
257
  make_video()
258
  if mode == 0:
259
+ text = "Press \"Start Regressing\" if your are happy with the training data. Regression takes ~30s."
260
  else:
261
  text = "Press \"Prepare Training Data\" before moving the sliders. You may also change the random seed."
262
  training_data_plot = 'training_data.png' if mode == 0 else None