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

[BUGFIX] Neural Gas gets prototype intiailizer from kwargs

Browse Source
This commit is contained in:
Alexander Engelsberger 2021-06-03 15:24:17 +02:00
parent 358f27257d
commit 0bc385fe7b
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prototorch/models/unsupervised.py
View File

@ -191,12 +191,10 @@ class NeuralGas(AbstractPrototypeModel):
self.hparams.setdefault("input_dim", 2)
self.hparams.setdefault("agelimit", 10)
self.hparams.setdefault("lm", 1)
self.hparams.setdefault("prototype_initializer",
ZerosInitializer(self.hparams.input_dim))
self.proto_layer = Components(
self.hparams.num_prototypes,
initializer=self.hparams.prototype_initializer)
initializer=kwargs.get("prototype_initializer"))
self.distance_layer = LambdaLayer(euclidean_distance)
self.energy_layer = NeuralGasEnergy(lm=self.hparams.lm)