Update example

2021-05-25 20:57:54 +02:00 · 2021-05-25 20:57:54 +02:00 · db965541fd
commit db965541fd
parent d091dea6a1
1 changed files with 4 additions and 3 deletions
--- a/examples/glvq_iris.py
+++ b/examples/glvq_iris.py
@ -24,10 +24,11 @@ if __name__ == "__main__":
                                               batch_size=150)

    # Hyperparameters
-    num_classes = 3
-    prototypes_per_class = 2
    hparams = dict(
-        distribution=(num_classes, prototypes_per_class),
+        distribution={
+            "num_classes": 3,
+            "prototypes_per_class": 4
+        },
        lr=0.01,
    )