Small fix on example script

2021-04-29 17:11:06 +02:00 · 2021-04-29 17:11:06 +02:00 · 8bad54fc2d
commit 8bad54fc2d
parent a16bebd0c4
1 changed files with 1 additions and 1 deletions
--- a/examples/glvq_iris_v1.py
+++ b/examples/glvq_iris_v1.py
@ -28,7 +28,7 @@ if __name__ == "__main__":
    )

    # Initialize the model
-    model = GLVQ(hparams, data=[x_train, y_train])
+    model = GLVQ(hparams)

    # Setup trainer
    trainer = pl.Trainer(