julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[BUGFIX] Fix lambda_matrix property in GMLVQ

Browse Source
This commit is contained in:
Jensun Ravichandran
2021-05-10 14:09:25 +02:00
parent c6e06ceaa4
commit ed03ab168e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
prototorch/models/glvq.py
View File

@@ -200,8 +200,8 @@ class GMLVQ(GLVQ):
@property @property
def lambda_matrix(self): def lambda_matrix(self):
omega = self.omega_layer.weight omega = self.omega_layer.weight # (latent_dim, input_dim)
lam = omega @ omega.T lam = omega.T @ omega
return lam.detach().cpu() return lam.detach().cpu()
def show_lambda(self): def show_lambda(self):