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

adjust omega_matrix property

Browse Source
This commit is contained in:
julius 2023-11-06 16:32:57 +01:00
parent 824dfced92
commit 1786031b4e
Signed by untrusted user who does not match committer: julius
GPG Key ID: 8AA3791362A8084A
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/prototorch/models/glvq.py
View File

@ -238,7 +238,8 @@ class GMLMLVQ(GLVQ):
""" """
# Parameters # Parameters
_omega: torch.Tensor _omega_0: torch.Tensor
_omega_1: torch.Tensor
def __init__(self, hparams, **kwargs): def __init__(self, hparams, **kwargs):
distance_fn = kwargs.pop("distance_fn", ML_omega_distance) distance_fn = kwargs.pop("distance_fn", ML_omega_distance)
@ -255,8 +256,8 @@ class GMLMLVQ(GLVQ):
self.mask_1 = masks[1] self.mask_1 = masks[1]
@property @property
def omega_matrix(self): def omega_matrices(self):
return self._omega.detach().cpu() return [self._omega_0.detach().cpu(), self._omega_1.detach().cpu()]
@property @property
def lambda_matrix(self): def lambda_matrix(self):