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 Relevance Learning Vector Quantization (GRLVQ)
- Generalized Matrix Learning Vector Quantization (GMLVQ) - Generalized Matrix Learning Vector Quantization (GMLVQ)
- Limited-Rank Matrix Learning Vector Quantization (LiRaMLVQ) - Limited-Rank Matrix Learning Vector Quantization (LiRaMLVQ)
- Learning Vector Quantization Multi-Layer Network (LVQMLN)
- Siamese GLVQ - Siamese GLVQ
- Neural Gas (NG) - Neural Gas (NG)
## Work in Progress ## Work in Progress
- Learning Vector Quantization Multi-Layer Network (LVQMLN)
- Classification-By-Components Network (CBC) - Classification-By-Components Network (CBC)
- Learning Vector Quantization 2.1 (LVQ2.1) - Learning Vector Quantization 2.1 (LVQ2.1)

View File

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

View File

@ -30,7 +30,7 @@ if __name__ == "__main__":
noise=1e-1)) noise=1e-1))
# Callbacks # 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) snan = StopOnNaN(model.proto_layer.components)
# Setup trainer # Setup trainer