Change constant to bound
Browse files
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
|
| 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
|
| 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
|
| 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 ===
|