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

Small changes

Browse Source
This commit is contained in:
Jensun Ravichandran
2021-05-17 19:37:42 +02:00
parent d812bb0620
commit 538256dcb7
3 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/cbc_iris.py
View File

@@ -33,7 +33,9 @@ if __name__ == "__main__":
# Callbacks
dvis = pt.models.VisCBC2D(data=(x_train, y_train),
title="CBC Iris Example")
title="CBC Iris Example",
resolution=300,
axis_off=True)
# Setup trainer
trainer = pl.Trainer(

2
examples/glvq_spiral.py
View File

@@ -30,7 +30,7 @@ if __name__ == "__main__":
noise=1e-1))
# Callbacks
vis = pt.models.VisGLVQ2D(train_ds, show_last_only=True, block=True)
vis = pt.models.VisGLVQ2D(train_ds, show_last_only=True)
snan = StopOnNaN(model.proto_layer.components)
# Setup trainer