nystrom-ncut 0.0.11__py3-none-any.whl → 0.0.12__py3-none-any.whl

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/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.dist-info → nystrom_ncut-0.0.12.dist-info}/METADATA

@@ -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.dist-info → nystrom_ncut-0.0.12.dist-info}/RECORD

@@ -2,13 +2,13 @@ __init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nystrom_ncut/__init__.py,sha256=JKfF6atok5T9V692RhlhgeRO5a2cN-bfAVa9irmTLfs,463
 nystrom_ncut/common.py,sha256=RMPQvg9R2s7V-q7zAStN9YCZt7gpc5Ut-KSKtvELBQ4,1934
 nystrom_ncut/propagation_utils.py,sha256=WeWKxRBm01ITILMgjsit5_fCe9oW1kJOPmAjjcmliMo,10340
-nystrom_ncut/visualize_utils.py,sha256=y7GvBbpbep7qeH6x8Rs8-0Fx57VohbQf9Wfl_vGgp3g,16853
+nystrom_ncut/visualize_utils.py,sha256=JkDyWML6k7k6S2Z7xnpbUvMWssEXcXqXu7gBy8wnids,16809
 nystrom_ncut/nystrom/__init__.py,sha256=4EpxD3Cmc8Fif4vo8DG-6FpTfCnNanD5zCZxK3WrMwQ,121
 nystrom_ncut/nystrom/distance_realization.py,sha256=MWSdfPfUnr7BdiKFkogjQvcGagvj7OzLQklnVp0fPx8,6000
 nystrom_ncut/nystrom/normalized_cut.py,sha256=_U3zrbe6V-5TQ4uWmqckxs2JTIhygQlnRDTFBI1ghD4,7194
-nystrom_ncut/nystrom/nystrom.py,sha256=VJPA17I8cVvjILUABJjkVA5kkXbTmHDyrtcWvu5xs-0,12571
-nystrom_ncut-0.0.11.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-nystrom_ncut-0.0.11.dist-info/METADATA,sha256=Lo3sy3uBMxHf-0JDsJuAzPc3A5MTYKagJ9_DSrF4AMk,6059
-nystrom_ncut-0.0.11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-nystrom_ncut-0.0.11.dist-info/top_level.txt,sha256=gM8IWWHYysIRTCvCTcdS4RShOyl9pxpylgSwPUZR2XM,22
-nystrom_ncut-0.0.11.dist-info/RECORD,,
+nystrom_ncut/nystrom/nystrom.py,sha256=nL-zxbEE_ygJEZEPmeNrUpVeffvxdrsTcbxFanFuXQY,12613
+nystrom_ncut-0.0.12.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+nystrom_ncut-0.0.12.dist-info/METADATA,sha256=pM-WT6Ly-IKYJ3DV2d-oOyc--K4VOyArB0sT5gHfHL4,6059
+nystrom_ncut-0.0.12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+nystrom_ncut-0.0.12.dist-info/top_level.txt,sha256=gM8IWWHYysIRTCvCTcdS4RShOyl9pxpylgSwPUZR2XM,22
+nystrom_ncut-0.0.12.dist-info/RECORD,,