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

Update LiRaMLVQ example script

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-09 20:54:40 +02:00
parent ca4c9da10a
commit 7d2af9c0ae
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/liramlvq_tecator.py
View File

@ -29,15 +29,9 @@ if __name__ == "__main__":
# Initialize the model # Initialize the model
model = pt.models.GMLVQ(hparams) model = pt.models.GMLVQ(hparams)
# Model summary
print(model)
# Callbacks # Callbacks
vis = pt.models.VisSiameseGLVQ2D(train_ds, border=0.1) vis = pt.models.VisSiameseGLVQ2D(train_ds, border=0.1)
# Namespace hook for the visualization to work
model.backbone = model.omega_layer
# Setup trainer # Setup trainer
trainer = pl.Trainer(max_epochs=200, callbacks=[vis]) trainer = pl.Trainer(max_epochs=200, callbacks=[vis])