nystrom-ncut 0.0.11__tar.gz → 0.0.12__tar.gz

{nystrom_ncut-0.0.11/src/nystrom_ncut.egg-info → nystrom_ncut-0.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nystrom_ncut
-Version: 0.0.11
+Version: 0.0.12
 Summary: Normalized Cut and Nyström Approximation
 Author-email: Huzheng Yang <huze.yann@gmail.com>, Wentinn Liao <wentinn.liao@gmail.com>
 Project-URL: Documentation, https://github.com/JophiArcana/Nystrom-NCUT/

{nystrom_ncut-0.0.11 → nystrom_ncut-0.0.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nystrom_ncut"
-version = "0.0.11"
+version = "0.0.12"
 authors = [
     { name = "Huzheng Yang", email = "huze.yann@gmail.com" },
     { name = "Wentinn Liao", email = "wentinn.liao@gmail.com" },

{nystrom_ncut-0.0.11 → nystrom_ncut-0.0.12}/src/nystrom_ncut/nystrom/nystrom.py

@@ -276,5 +276,7 @@ def solve_eig(
     eigen_vector = eigen_vector[:, indices]
 
     # correct the random rotation (flipping sign) of eigenvectors
-    eigen_vector = eigen_vector * torch.sum(eigen_vector, dim=0).sign()
+    sign = torch.sum(eigen_vector, dim=0).sign()
+    sign[sign == 0] = 1.0
+    eigen_vector = eigen_vector * sign
     return eigen_vector, eigen_value

{nystrom_ncut-0.0.11 → nystrom_ncut-0.0.12}/src/nystrom_ncut/visualize_utils.py

@@ -195,8 +195,8 @@ def rgb_from_cosine_tsne_3d(
         )
         perplexity = num_sample // 2
 
-    def cosine_to_rbf(X: torch.Tensor) -> torch.Tensor:                                 # [B... x N x 3]
-        dr = DistanceRealization(n_components=3, num_sample=20000, distance="cosine", eig_solver="lobpcg")
+    def cosine_to_rbf(X: torch.Tensor) -> torch.Tensor:
+        dr = DistanceRealization(n_components=3, num_sample=20000, distance="cosine", eig_solver="svd_lowrank")
         return dr.fit_transform(X)
 
     def rgb_from_cosine(X_3d: torch.Tensor, q: float) -> torch.Tensor:

{nystrom_ncut-0.0.11 → nystrom_ncut-0.0.12/src/nystrom_ncut.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nystrom_ncut
-Version: 0.0.11
+Version: 0.0.12
 Summary: Normalized Cut and Nyström Approximation
 Author-email: Huzheng Yang <huze.yann@gmail.com>, Wentinn Liao <wentinn.liao@gmail.com>
 Project-URL: Documentation, https://github.com/JophiArcana/Nystrom-NCUT/