nystrom-ncut 0.3.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. nystrom_ncut/kernel/kernel_ncut.py +6 -1
  2. {nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/METADATA +1 -1
  3. {nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/RECORD +6 -6
  4. {nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/LICENSE +0 -0
  5. {nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/WHEEL +0 -0
  6. {nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/top_level.txt +0 -0
nystrom_ncut/kernel/kernel_ncut.py CHANGED
@@ -28,10 +28,12 @@ class KernelNCutBaseTransformer(OnlineTorchTransformerMixin):
28
28
  self.affinity_focal_gamma = affinity_focal_gamma
29
29
 
30
30
  # Anchor matrices
31
+ self.anchor_count: int = None # n
31
32
  self.W: torch.Tensor = None # [... x d x kernel_dim]
32
33
  self.kernelized_anchor: torch.Tensor = None # [... x n x (2 * kernel_dim)]
33
34
 
34
35
  # Updated matrices
36
+ self.total_count: int = None # m
35
37
  self.r: torch.Tensor = None # [... x (2 * kernel_dim)]
36
38
  self.transform_matrix: torch.Tensor = None # [... x (2 * kernel_dim) x n_components]
37
39
  self.eigenvalues_: torch.Tensor = None # [... x n_components]
@@ -42,10 +44,12 @@ class KernelNCutBaseTransformer(OnlineTorchTransformerMixin):
42
44
  _, S, V = torch.svd_lowrank(torch.nan_to_num(
43
45
  normalized_kernelized_anchor, nan=0.0,
44
46
  ), q=self.n_components) # [... x n_components], [... x (2 * kernel_dim) x n_components]
45
- self.transform_matrix = V * torch.nan_to_num(1 / S, posinf=0.0, neginf=0.0)[..., None, :] # [... x (2 * kernel_dim) x n_components]
47
+ S = S * (self.total_count / self.anchor_count) ** 0.5
48
+ self.transform_matrix = V * torch.nan_to_num(1 / S, posinf=0.0, neginf=0.0)[..., None, :] # [... x (2 * kernel_dim) x n_components]
46
49
  self.eigenvalues_ = S ** 2
47
50
 
48
51
  def fit(self, features: torch.Tensor) -> "KernelNCutBaseTransformer":
52
+ self.anchor_count = self.total_count = features.shape[-2]
49
53
  d = features.shape[-1]
50
54
  scale = get_normalization_factor(features) * (self.affinity_focal_gamma ** 0.5) # [...]
51
55
  self.W = torch.randn((*features.shape[:-2], d, self.kernel_dim)) / scale[..., None, None] # [... x d x kernel_dim]
@@ -60,6 +64,7 @@ class KernelNCutBaseTransformer(OnlineTorchTransformerMixin):
60
64
  return self
61
65
 
62
66
  def update(self, features: torch.Tensor) -> torch.Tensor:
67
+ self.total_count += features.shape[-2]
63
68
  W_features = features @ self.W # [... x m x kernel_dim]
64
69
  kernelized_features = torch.cat((
65
70
  torch.cos(W_features),
{nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: nystrom_ncut
3
- Version: 0.3.4
3
+ Version: 0.3.5
4
4
  Summary: Normalized Cut and Nyström Approximation
5
5
  Author-email: Huzheng Yang <huze.yann@gmail.com>, Wentinn Liao <wentinn.liao@gmail.com>
6
6
  Project-URL: Documentation, https://github.com/JophiArcana/Nystrom-NCUT/
{nystrom_ncut-0.3.4.dist-info → nystrom_ncut-0.3.5.dist-info}/RECORD RENAMED
@@ -6,7 +6,7 @@ nystrom_ncut/global_settings.py,sha256=TckHuF8geWM2ofd99jBupHD3TQdeEB583_3pdIVRU
6
6
  nystrom_ncut/sampling_utils.py,sha256=lPWWNcDMBIvK9MmtNkVUeokeCOH9P1Fm8TZZ4sjlCpM,9037
7
7
  nystrom_ncut/visualize_utils.py,sha256=A1qmL8eNZjtvOOlyl9KIeObnpPVUGZVsCB9QBRV7n9I,22762
8
8
  nystrom_ncut/kernel/__init__.py,sha256=pvJ3tFukmlNZqw8VUB_iKPY9gbchUAlNkmnRCZcp0uU,44
9
- nystrom_ncut/kernel/kernel_ncut.py,sha256=u2oIu-SmNGn9tg-SXBJY4G9WGJ0SeGSnyZcvX8DGAE0,5276
9
+ nystrom_ncut/kernel/kernel_ncut.py,sha256=fEOKbRoNSAsdWEsUaw2-2SEoscaJcMy0PaxfH6Va3Sg,5567
10
10
  nystrom_ncut/nystrom/__init__.py,sha256=NHse-dW4nTo9wJUEJ6G4_Gw8uAi2vP6Hxm9aeBWxmqc,49
11
11
  nystrom_ncut/nystrom/distance_realization.py,sha256=kvPS-jGUn85MRJx-Dh2IZJ3IwvavRDCbXq6wh_aEBxc,5684
12
12
  nystrom_ncut/nystrom/normalized_cut.py,sha256=BD1F9Wz1BXbTGC-AVwT4IGmsPp334z-7jE9CuwhpNjY,7415
@@ -14,8 +14,8 @@ nystrom_ncut/nystrom/nystrom_utils.py,sha256=5cMoF8UFgi_N-nEzbSQqVGhuep_eOn-FzEr
14
14
  nystrom_ncut/transformer/__init__.py,sha256=2FJEG9CXavfDDdDk1i9OOGkd8uSOHMkP8LBH49nnPnM,138
15
15
  nystrom_ncut/transformer/axis_align.py,sha256=j3LlAPrp8O_jQAlwZz-gu3D7n_wICEJranye-YK5wvA,4880
16
16
  nystrom_ncut/transformer/transformer_mixin.py,sha256=9wYdWknnJP7jijz70lRAyDn_kpC9aWwYN7Pbt5Mf6gQ,2012
17
- nystrom_ncut-0.3.4.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
18
- nystrom_ncut-0.3.4.dist-info/METADATA,sha256=I3V55oiDNJl4hw9v7TdEb1ibmJWtU5Ife9KwJ03rJPQ,6058
19
- nystrom_ncut-0.3.4.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
20
- nystrom_ncut-0.3.4.dist-info/top_level.txt,sha256=gM8IWWHYysIRTCvCTcdS4RShOyl9pxpylgSwPUZR2XM,22
21
- nystrom_ncut-0.3.4.dist-info/RECORD,,
17
+ nystrom_ncut-0.3.5.dist-info/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
18
+ nystrom_ncut-0.3.5.dist-info/METADATA,sha256=h_tVSIbUZdkM2jxFshTodlinpcwKHU0ZIWxnXA-Qy2k,6058
19
+ nystrom_ncut-0.3.5.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
20
+ nystrom_ncut-0.3.5.dist-info/top_level.txt,sha256=gM8IWWHYysIRTCvCTcdS4RShOyl9pxpylgSwPUZR2XM,22
21
+ nystrom_ncut-0.3.5.dist-info/RECORD,,