Showing initial point cloud

Xiaoyiyong555 · Mar 15, 2022 · 74c51c7 · 74c51c7
1 parent 0d3950e
commit 74c51c7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torch_topological/examples/summary_statistics.py b/torch_topological/examples/summary_statistics.py
@@ -36,6 +36,7 @@ def main(args):
 
     if args.single:
         X = sample_from_unit_cube(n=n, d=2)
+        Y = X.clone()
     else:
         X = sample_from_disk(n=n, r=0.5, R=0.6)
         Y = sample_from_disk(n=n, r=0.9, R=1.0)
@@ -73,7 +74,8 @@ def main(args):
     X = X.detach().numpy()
 
     if args.single:
-        plt.scatter(X[:, 0], X[:, 1], label='Output')
+        plt.scatter(X[:, 0], X[:, 1], label='Result')
+        plt.scatter(Y[:, 0], Y[:, 1], label='Initial')
     else:
         plt.scatter(X[:, 0], X[:, 1], label='Source')
         plt.scatter(Y[:, 0], Y[:, 1], label='Target')