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

[REFACTOR] Remove prototype_initializer function from GLVQ

Browse Source 
Fixes #9
This commit is contained in:
Alexander Engelsberger
2021-06-03 15:15:22 +02:00
parent bda88149d4
commit 358f27257d
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
prototorch/models/glvq.py
View File

@@ -34,9 +34,10 @@ class GLVQ(AbstractPrototypeModel):
transfer_fn = get_activation(self.hparams.transfer_fn)
# Layers
prototype_initializer = kwargs.get("prototype_initializer", None)
self.proto_layer = LabeledComponents(
distribution=self.hparams.distribution,
initializer=self.prototype_initializer(**kwargs))
initializer=prototype_initializer)
self.distance_layer = LambdaLayer(distance_fn)
self.transfer_layer = LambdaLayer(transfer_fn)
@@ -47,9 +48,6 @@ class GLVQ(AbstractPrototypeModel):
self.optimizer = kwargs.get("optimizer", torch.optim.Adam)
def prototype_initializer(self, **kwargs):
return kwargs.get("prototype_initializer", None)
@property
def prototype_labels(self):
return self.proto_layer.component_labels.detach().cpu()