Update.

[pytorch.git] / miniflow.py

diff --git a/miniflow.py b/miniflow.py
index e4f5945..b5c8cb4 100755 (executable)
--- 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()