nystrom-ncut 0.3.1__tar.gz → 0.3.3__tar.gz

{nystrom_ncut-0.3.1/src/nystrom_ncut.egg-info → nystrom_ncut-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nystrom_ncut
-Version: 0.3.1
+Version: 0.3.3
 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.3.1 → nystrom_ncut-0.3.3}/pyproject.toml

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

{nystrom_ncut-0.3.1 → nystrom_ncut-0.3.3}/src/nystrom_ncut/nystrom/normalized_cut.py

@@ -54,10 +54,10 @@ class LaplacianKernel(OnlineKernel):
             self.A,
             num_eig=d + 1,  # d * (d + 3) // 2 + 1,
             eig_solver=self.eig_solver,
-        )                                                                           # [... x n x (d + 1)], [... x (d + 1)]
-        self.Ainv = U @ torch.diag_embed(1 / L) @ U.mT                              # [... x n x n]
-        self.a_r = torch.where(self.anchor_mask, torch.inf, torch.sum(self.A, dim=-1))  # [... x n]
-        self.b_r = torch.zeros_like(self.a_r)                                       # [... x n]
+        )                                                                                           # [... x n x (d + 1)], [... x (d + 1)]
+        self.Ainv = U @ torch.nan_to_num(torch.diag_embed(1 / L), posinf=0.0, neginf=0.0) @ U.mT    # [... x n x n]
+        self.a_r = torch.where(self.anchor_mask, torch.inf, torch.sum(self.A, dim=-1))              # [... x n]
+        self.b_r = torch.zeros_like(self.a_r)                                                       # [... x n]
 
     def _affinity(self, features: torch.Tensor) -> torch.Tensor:
         B = torch.where(self.anchor_mask[..., None], 0.0, affinity_from_features(

{nystrom_ncut-0.3.1 → nystrom_ncut-0.3.3}/src/nystrom_ncut/nystrom/nystrom_utils.py

@@ -269,7 +269,7 @@ def solve_eig(
     bsz: int = A.shape[0]
 
     A = A + eig_value_buffer * torch.eye(A.shape[-1], device=A.device)
-
+    num_eig = min(A.shape[-1], num_eig)
     # compute eigenvectors
     if eig_solver == "svd_lowrank":  # default
         # only top q eigenvectors, fastest

{nystrom_ncut-0.3.1 → nystrom_ncut-0.3.3/src/nystrom_ncut.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nystrom_ncut
-Version: 0.3.1
+Version: 0.3.3
 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/