[BUGFIX] examples/gng_iris.py works again

2021-06-14 20:29:31 +02:00
parent 4eafe88dc4
commit d2856383e2
2 changed files with 3 additions and 2 deletions
--- a/examples/gng_iris.py
+++ b/examples/gng_iris.py
@@ -29,7 +29,7 @@ if __name__ == "__main__":
    # Initialize the model
    model = pt.models.GrowingNeuralGas(
        hparams,
-        prototype_initializer=pt.components.Zeros(2),
+        prototypes_initializer=pt.initializers.ZCI(2),
    )

    # Compute intermediate input and output sizes