Small changes

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

View File

@ -28,12 +28,12 @@ The plugin should then be available for use in your Python environment as
- Generalized Relevance Learning Vector Quantization (GRLVQ)
- Generalized Matrix Learning Vector Quantization (GMLVQ)
- Limited-Rank Matrix Learning Vector Quantization (LiRaMLVQ)
- Learning Vector Quantization Multi-Layer Network (LVQMLN)
- Siamese GLVQ
- Neural Gas (NG)
## Work in Progress
- Learning Vector Quantization Multi-Layer Network (LVQMLN)
- Classification-By-Components Network (CBC)
- Learning Vector Quantization 2.1 (LVQ2.1)

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(

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