phess2 commited on
Commit
346b750
·
1 Parent(s): 75c2f83

Change constant to bound

Browse files
Files changed (1) hide show
  1. figures/reproduce_figures.py +3 -3
figures/reproduce_figures.py CHANGED
@@ -122,7 +122,7 @@ def create_figure_2(MLP_highlight_points, transformer_highlight_points):
122
  label=label, markersize=10)
123
  )
124
 
125
- ax.set_xlabel("Lipschitz constant", fontsize=label_fontsize)
126
  if show_ylabel:
127
  ax.set_ylabel("Validation loss", fontsize=label_fontsize)
128
  ax.set_xscale("log")
@@ -345,7 +345,7 @@ def create_figure_4(highlight_points, results_df):
345
  ax0.set_xscale('log')
346
  ax0.set_yscale('log')
347
  ax0.minorticks_off()
348
- ax0.set_xlabel('Lipschitz constant', fontsize=label_fontsize)
349
  ax0.set_ylabel('Validation loss', fontsize=label_fontsize)
350
 
351
  class CustomLogFormatter(LogFormatter):
@@ -398,7 +398,7 @@ def create_figure_4(highlight_points, results_df):
398
  legend_labels.append(tech_map.get(technique, technique))
399
 
400
  ax1.set_ylim(0.545, upper_threshold)
401
- ax1.set_xlabel('Lipschitz constant', fontsize=label_fontsize)
402
  ax1.set_ylabel('Best validation accuracy', fontsize=label_fontsize)
403
 
404
  # === RIGHT PLOT ===
 
122
  label=label, markersize=10)
123
  )
124
 
125
+ ax.set_xlabel("Lipschitz bound", fontsize=label_fontsize)
126
  if show_ylabel:
127
  ax.set_ylabel("Validation loss", fontsize=label_fontsize)
128
  ax.set_xscale("log")
 
345
  ax0.set_xscale('log')
346
  ax0.set_yscale('log')
347
  ax0.minorticks_off()
348
+ ax0.set_xlabel('Lipschitz bound', fontsize=label_fontsize)
349
  ax0.set_ylabel('Validation loss', fontsize=label_fontsize)
350
 
351
  class CustomLogFormatter(LogFormatter):
 
398
  legend_labels.append(tech_map.get(technique, technique))
399
 
400
  ax1.set_ylim(0.545, upper_threshold)
401
+ ax1.set_xlabel('Lipschitz bound', fontsize=label_fontsize)
402
  ax1.set_ylabel('Best validation accuracy', fontsize=label_fontsize)
403
 
404
  # === RIGHT PLOT ===