fix square bug

epfl-lts2 · May 14, 2020 · 2e80742 · 2e80742 · GautamSridhar · May 15, 2020
1 parent 27ed937
commit 2e80742
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygsp/graphs/learned.py b/pygsp/graphs/learned.py
@@ -531,11 +531,11 @@ def __init__(self, X, a=None, b=None, k=10, kk=None, sparse=True, rel_edge=1e-5,
             if kk is None:
                 kk = 3*k
             neighbors, distances = nearest_neighbor(X, k=kk, **param_nn)
-            Z = sparse_distance_matrix(neighbors, distances)
+            Z = sparse_distance_matrix(neighbors, distances**2)
             edge_mask = Z>0
             Zp = distances[:,1:]
         else:
-            Z = distanz(X.transpose())
+            Z = distanz(X.transpose())**2
             Zp = Z
             edge_mask = None
         if a is None and b is None: