fix: rename hparam output_dim → latent_dim in SiameseGMLVQ

2022-03-29 15:24:42 +02:00
parent 40bd7ed380
commit fab786a07e
2 changed files with 2 additions and 2 deletions
--- a/docs/source/tutorial.ipynb
+++ b/docs/source/tutorial.ipynb
@@ -422,7 +422,7 @@
    "trained_model = pt.models.GLVQ.load_from_checkpoint(\"./checkpoints/glvq_iris.ckpt\", strict=False)\n",
    "model = pt.models.SiameseGMLVQ(\n",
    "    dict(input_dim=2,\n",
-    "         output_dim=2,\n",
+    "         latent_dim=2,\n",
    "         distribution=(3, 2),\n",
    "         proto_lr=0.0001,\n",
    "         bb_lr=0.0001),\n",