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

nystrom_ncut/nystrom/distance_realization.py

@@ -1,3 +1,5 @@
+from typing import Tuple
+
 import torch
 
 from .nystrom import (
@@ -125,3 +127,15 @@ class DistanceRealization(OnlineNystromSubsampleFit):
             chunk_size=chunk_size,
         )
         self.distance: DistanceOptions = distance
+
+    def fit_transform(
+        self,
+        features: torch.Tensor,
+        precomputed_sampled_indices: torch.Tensor = None,
+    ) -> torch.Tensor:
+        V, L = OnlineNystromSubsampleFit.fit_transform(self, features, precomputed_sampled_indices)
+        return V * (L ** 0.5)
+
+    def transform(self, features: torch.Tensor = None) -> torch.Tensor:
+        V, L = OnlineNystromSubsampleFit.transform(features)
+        return V * (L ** 0.5)

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

@@ -11,6 +11,9 @@ from .common import (
     quantile_min_max,
     quantile_normalize,
 )
+from .nystrom import (
+    DistanceRealization,
+)
 from .propagation_utils import (
     run_subgraph_sampling,
     extrapolate_knn,
@@ -192,18 +195,9 @@ def rgb_from_cosine_tsne_3d(
         )
         perplexity = num_sample // 2
 
-
-    def cosine_to_rbf(X: torch.Tensor) -> torch.Tensor:                                 # [B... x N x 3]
-        normalized_X = X / torch.norm(X, p=2, dim=-1, keepdim=True)                     # [B... x N x 3]
-        D = 1 - normalized_X @ normalized_X.mT                                          # [B... x N x N]
-
-        G = (D[..., :1, 1:] ** 2 + D[..., 1:, :1] ** 2 - D[..., 1:, 1:] ** 2) / 2       # [B... x (N - 1) x (N - 1)]
-        L, V = torch.linalg.eigh(G)                                                     # [B... x (N - 1)], [B... x (N - 1) x (N - 1)]
-        sqrtG = V[..., -3:] * (L[..., None, -3:] ** 0.5)                                # [B... x (N - 1) x 3]
-
-        Y = torch.cat((torch.zeros_like(sqrtG[..., :1, :]), sqrtG), dim=-2)             # [B... x N x 3]
-        Y = Y - torch.mean(Y, dim=-2, keepdim=True)
-        return Y
+    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:
         return rgb_from_3d_rgb_cube(cosine_to_rbf(X_3d), q=q)
@@ -379,7 +373,6 @@ def rotate_rgb_cube(rgb, position=1):
 
 def rgb_from_3d_rgb_cube(X_3d, q=0.95):
     """convert 3D t-SNE to RGB color space
-
     Args:
         X_3d (torch.Tensor): 3D t-SNE embedding, shape (n_samples, 3)
         q (float): quantile, default 0.95
@@ -389,10 +382,10 @@ def rgb_from_3d_rgb_cube(X_3d, q=0.95):
     """
     assert X_3d.shape[1] == 3, "input should be (n_samples, 3)"
     assert len(X_3d.shape) == 2, "input should be (n_samples, 3)"
-    rgb = []
-    for i in range(3):
-        rgb.append(quantile_normalize(X_3d[:, i], q=q))
-    rgb = torch.stack(rgb, dim=-1)
+    rgb = torch.stack([
+        quantile_normalize(x, q=q)
+        for x in torch.unbind(X_3d, dim=1)
+    ], dim=-1)
     return rgb
 
 

{nystrom_ncut-0.0.10.dist-info → nystrom_ncut-0.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nystrom_ncut
-Version: 0.0.10
+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.12.dist-info/RECORD

@@ -0,0 +1,14 @@
+__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=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=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,,

nystrom_ncut-0.0.10.dist-info/RECORD

@@ -1,14 +0,0 @@
-__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=Z_bcoxwmWpTxhQ_yoAXqTnYDf269IuT0b0Sm2EVQpRw,17422
-nystrom_ncut/nystrom/__init__.py,sha256=4EpxD3Cmc8Fif4vo8DG-6FpTfCnNanD5zCZxK3WrMwQ,121
-nystrom_ncut/nystrom/distance_realization.py,sha256=8AWUlZKZEPfhQHxYTZt0uzKedVp8ZB1wb__7M2Fy-Eo,5529
-nystrom_ncut/nystrom/normalized_cut.py,sha256=_U3zrbe6V-5TQ4uWmqckxs2JTIhygQlnRDTFBI1ghD4,7194
-nystrom_ncut/nystrom/nystrom.py,sha256=VJPA17I8cVvjILUABJjkVA5kkXbTmHDyrtcWvu5xs-0,12571
-nystrom_ncut-0.0.10.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-nystrom_ncut-0.0.10.dist-info/METADATA,sha256=sqs2WHdNbJeT5zvlq_WWHHRvHTz1mHVbDL3PsE1NMBI,6059
-nystrom_ncut-0.0.10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-nystrom_ncut-0.0.10.dist-info/top_level.txt,sha256=gM8IWWHYysIRTCvCTcdS4RShOyl9pxpylgSwPUZR2XM,22
-nystrom_ncut-0.0.10.dist-info/RECORD,,