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
Show Stats 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
def lambda_matrix(self):
omega = self.omega_layer.weight
lam = omega @ omega.T
omega = self.omega_layer.weight # (latent_dim, input_dim)
lam = omega.T @ omega
return lam.detach().cpu()
def show_lambda(self):