Update _layout.py

Droxef · Jun 11, 2019 · 555e636 · 555e636
1 parent 8ce5bde
commit 555e636
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))