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

Fix typo

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-04-23 17:38:29 +02:00
parent 466bbe4c63
commit 281009ce82
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/ng_iris.py
View File

@ -1,4 +1,4 @@
"""CBC example using the Iris dataset.""" """Neural Gas example using the Iris dataset."""
import numpy as np import numpy as np
import pytorch_lightning as pl import pytorch_lightning as pl
@ -80,7 +80,7 @@ if __name__ == "__main__":
nclasses=1, nclasses=1,
prototypes_per_class=30, prototypes_per_class=30,
prototype_initializer="rand", prototype_initializer="rand",
lr=0.01, lr=0.1,
) )
# Initialize the model # Initialize the model