X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=miniflow.py;h=b5c8cb430bceab57c05136b4c26e534b56ab257e;hp=e4f594588f441775d63885e691bc4a19ce271955;hb=7080726691be9341436db5a664778679600c5f62;hpb=78b4450fc60a5db62bc8ed50ec54e255a60f24e2 diff --git a/miniflow.py b/miniflow.py index e4f5945..b5c8cb4 100755 --- a/miniflow.py +++ b/miniflow.py @@ -202,26 +202,25 @@ fig.savefig(filename, bbox_inches='tight') ###################################################################### -z = torch.empty(200).normal_() -z = z[(z > -3) * (z < 3)] +# z = torch.empty(200).normal_() +# z = z[(z > -3) * (z < 3)] -x = model.invert(z) +# x = model.invert(z) -fig = plt.figure() -ax = fig.add_subplot(1, 1, 1) -ax.set_xlim(-4.5, 4.5) -ax.set_ylim(-0.1, 1.1) -lines = list(([(x_in.item(), 0), (x_out.item(), 0.5)]) for (x_in, x_out) in zip(x, z)) -lc = mc.LineCollection(lines, color = 'tab:red', linewidth = 0.1) -ax.add_collection(lc) -# ax.axis('off') - -# ax.fill_between(input, 0.52, mu_N * 0.2 + 0.52, color = green_dist) -# ax.fill_between(input, -0.30, mu * 0.2 - 0.30, color = green_dist) +# fig = plt.figure() +# ax = fig.add_subplot(1, 1, 1) +# ax.set_xlim(-4.5, 4.5) +# ax.set_ylim(-0.1, 1.1) +# lines = list(([(x_in.item(), 0), (x_out.item(), 0.5)]) for (x_in, x_out) in zip(x, z)) +# lc = mc.LineCollection(lines, color = 'tab:red', linewidth = 0.1) +# ax.add_collection(lc) +# # ax.axis('off') -filename = 'miniflow_synth.pdf' -print(f'Saving {filename}') -fig.savefig(filename, bbox_inches='tight') +# # ax.fill_between(input, 0.52, mu_N * 0.2 + 0.52, color = green_dist) +# # ax.fill_between(input, -0.30, mu * 0.2 - 0.30, color = green_dist) -# plt.show() +# filename = 'miniflow_synth.pdf' +# print(f'Saving {filename}') +# fig.savefig(filename, bbox_inches='tight') +# # plt.show()