nystrom-ncut 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

nystrom_ncut/ncut_pytorch.py

@@ -173,9 +173,10 @@ class NCUT(OnlineNystrom):
         else:
             sampled_indices = run_subgraph_sampling(
                 features,
-                num_sample=self.num_sample,
+                self.num_sample,
                 sample_method=self.sample_method,
             )
+        sampled_indices = torch.sort(sampled_indices).values
         sampled_features = features[sampled_indices]
         OnlineNystrom.fit(self, sampled_features)
 

nystrom_ncut/propagation_utils.py

@@ -11,7 +11,7 @@ from .common import ceildiv, lazy_normalize
 @torch.no_grad()
 def run_subgraph_sampling(
     features: torch.Tensor,
-    num_sample: int = 300,
+    num_sample: int,
     max_draw: int = 1000000,
     sample_method: Literal["farthest", "random"] = "farthest",
 ):
@@ -272,7 +272,7 @@ def propagate_eigenvectors(
     # sample subgraph
     subgraph_indices = run_subgraph_sampling(
         features,
-        num_sample=num_sample,
+        num_sample,
         sample_method=sample_method,
     )
 

nystrom_ncut/visualize_utils.py

@@ -34,7 +34,7 @@ def _rgb_with_dimensionality_reduction(
 ) -> Tuple[torch.Tensor, torch.Tensor]:
     subgraph_indices = run_subgraph_sampling(
         features,
-        num_sample=num_sample,
+        num_sample,
         sample_method="farthest",
     )
 

{nystrom_ncut-0.0.4.dist-info → nystrom_ncut-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nystrom_ncut
-Version: 0.0.4
+Version: 0.0.5
 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.5.dist-info/RECORD

@@ -0,0 +1,11 @@
+nystrom_ncut/__init__.py,sha256=Cww-_OsyQHLKpgw_Wh28_tUOvIMMr7Ey8w-tH7v99xQ,452
+nystrom_ncut/common.py,sha256=qdR_JwknT9H1Cv5LopwdwZfORFx-O8MLiRI6ZF1Qohc,558
+nystrom_ncut/ncut_pytorch.py,sha256=wRQXUPBOW2_vutocKf0J19HrFVkBYQePAYUEfotLfx4,11701
+nystrom_ncut/nystrom.py,sha256=HbwON9pLW3gEZvOmbDJwkQNHolOo1EBvwBPeh2p2uJE,8833
+nystrom_ncut/propagation_utils.py,sha256=mD6rZ_mwYjYXs1cp5ZaTK0FrJ4YhyCdoIUrdGRP9k-M,12119
+nystrom_ncut/visualize_utils.py,sha256=QmBatlX7Q-ZWF_iJ1zFDnPHFuofz3tCmtoNeeoMPw3U,18558
+nystrom_ncut-0.0.5.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+nystrom_ncut-0.0.5.dist-info/METADATA,sha256=n9zlRYBD02k478INScrj9V9rZ1mhXTylcMjkmQDgl1A,6058
+nystrom_ncut-0.0.5.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+nystrom_ncut-0.0.5.dist-info/top_level.txt,sha256=j7g_j0S048EvguFFnGgD5Ewd3r2H6klsxd5A4dd-wHw,13
+nystrom_ncut-0.0.5.dist-info/RECORD,,

nystrom_ncut-0.0.4.dist-info/RECORD

@@ -1,11 +0,0 @@
-nystrom_ncut/__init__.py,sha256=Cww-_OsyQHLKpgw_Wh28_tUOvIMMr7Ey8w-tH7v99xQ,452
-nystrom_ncut/common.py,sha256=qdR_JwknT9H1Cv5LopwdwZfORFx-O8MLiRI6ZF1Qohc,558
-nystrom_ncut/ncut_pytorch.py,sha256=8LfznDwhq-WL_vQxbFBFLSzymg9SEDti_zzf9QQLnrA,11651
-nystrom_ncut/nystrom.py,sha256=HbwON9pLW3gEZvOmbDJwkQNHolOo1EBvwBPeh2p2uJE,8833
-nystrom_ncut/propagation_utils.py,sha256=pigecB0rAmlbCoMNb8zhCyyNwh3QzkxXEnaBsDRE_ns,12136
-nystrom_ncut/visualize_utils.py,sha256=oNaDz_Xn12g3knEZZTb-QWVN-wTrnCNE5gn9cu8Xl_U,18569
-nystrom_ncut-0.0.4.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-nystrom_ncut-0.0.4.dist-info/METADATA,sha256=dog8rG5_vF31_SJS90ruUeJwnrs3bM635m7KSPLht78,6058
-nystrom_ncut-0.0.4.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
-nystrom_ncut-0.0.4.dist-info/top_level.txt,sha256=j7g_j0S048EvguFFnGgD5Ewd3r2H6klsxd5A4dd-wHw,13
-nystrom_ncut-0.0.4.dist-info/RECORD,,