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

[Bugfix] Fix classcount in LIRAMLVQ example

Browse Source
This commit is contained in:
Alexander Engelsberger 2021-05-31 11:48:23 +02:00
parent 2a218c0ede
commit 8f4d66edf1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/liramlvq_tecator.py
View File

@ -27,8 +27,8 @@ if __name__ == "__main__":
# Hyperparameters # Hyperparameters
hparams = dict( hparams = dict(
distribution={ distribution={
"num_classes": 3, "num_classes": 2,
"prototypes_per_class": 4 "prototypes_per_class": 2
}, },
input_dim=100, input_dim=100,
latent_dim=2, latent_dim=2,