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

Update example

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-25 20:57:54 +02:00
parent d091dea6a1
commit db965541fd
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/glvq_iris.py
View File

@ -24,10 +24,11 @@ if __name__ == "__main__":
batch_size=150) batch_size=150)
# Hyperparameters # Hyperparameters
num_classes = 3
prototypes_per_class = 2
hparams = dict( hparams = dict(
distribution=(num_classes, prototypes_per_class), distribution={
"num_classes": 3,
"prototypes_per_class": 4
},
lr=0.01, lr=0.01,
) )