diff --git a/mode_converter.ipynb b/mode_converter.ipynb index f20f76c..36d6890 100644 --- a/mode_converter.ipynb +++ b/mode_converter.ipynb @@ -38,6 +38,10 @@ "\n", "mp.verbosity(1)\n", "\n", + "#setup a results folder\n", + "if not os.path.exists(\"results\"):\n", + " os.makedirs(\"results\")\n", + "\n", "subs_index = 3.4\n", "subs_perm = subs_index**2\n", "\n", diff --git a/mode_converter.py b/mode_converter.py index 69232bf..3958b0c 100644 --- a/mode_converter.py +++ b/mode_converter.py @@ -17,6 +17,10 @@ import os mp.verbosity(1) +#setup a results folder +if not os.path.exists("results"): + os.makedirs("results") + subs_index = 3.4 subs_perm = subs_index**2 @@ -267,9 +271,6 @@ def f(v, gradient, cur_beta): circ2 = Circle((2,2), minimum_length/2) ax2.add_patch(circ2) - if not os.path.exists("results"): - os.makedirs("results") - plt.savefig("results/iter_{}.pdf".format(cur_iter[0]+1)) cur_iter[0] += cur_iter[0] + 1 @@ -308,7 +309,6 @@ if optimization_start: cur_beta = cur_beta * beta_scale #update beta print("current beta: ", cur_beta) - ######RESULTS###### #plot the figure of merit over iterations plt.figure()