Merge pull request epfl-lts2#54 from rodrigo-pena/fix-laplacian-eigen…

…maps Compute the correct number of eigenvectors for Laplacian eigenmaps.
Droxef · Jun 12, 2019 · 519fad1 · 519fad1
2 parents 8ce5bde + 555e636
commit 519fad1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygsp/graphs/_layout.py b/pygsp/graphs/_layout.py
@@ -92,10 +92,10 @@ def set_coordinates(self, kind='spring', **kwargs):
                 self.coords[i] = self.info['com_coords'][comm_idx] + \
                     comm_rad * self.coords[i]
         elif kind == 'laplacian_eigenmap2D':
-            self.compute_fourier_basis(n_eigenvectors=2)
+            self.compute_fourier_basis(n_eigenvectors=3)
             self.coords = self.U[:, 1:3]
         elif kind == 'laplacian_eigenmap3D':
-            self.compute_fourier_basis(n_eigenvectors=3)
+            self.compute_fourier_basis(n_eigenvectors=4)
             self.coords = self.U[:, 1:4]
         else:
             raise ValueError('Unexpected argument kind={}.'.format(kind))