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

fix(vis): plot prototypes after data

Browse Source
This commit is contained in:
Jensun Ravichandran 2022-06-14 19:59:13 +02:00
parent 46ec7b07d7
commit 94730f492b
No known key found for this signature in database
GPG Key ID: 7612C0CAB643D921
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/models/vis.py
View File

@ -169,13 +169,13 @@ class VisGLVQ2D(Vis2DAbstract):
plabels = pl_module.prototype_labels plabels = pl_module.prototype_labels
x_train, y_train = self.x_train, self.y_train x_train, y_train = self.x_train, self.y_train
ax = self.setup_ax() ax = self.setup_ax()
self.plot_protos(ax, protos, plabels)
if x_train is not None: if x_train is not None:
self.plot_data(ax, x_train, y_train) self.plot_data(ax, x_train, y_train)
mesh_input, xx, yy = mesh2d(np.vstack([x_train, protos]), mesh_input, xx, yy = mesh2d(np.vstack([x_train, protos]),
self.border, self.resolution) self.border, self.resolution)
else: else:
mesh_input, xx, yy = mesh2d(protos, self.border, self.resolution) mesh_input, xx, yy = mesh2d(protos, self.border, self.resolution)
self.plot_protos(ax, protos, plabels)
_components = pl_module.proto_layer._components _components = pl_module.proto_layer._components
mesh_input = torch.from_numpy(mesh_input).type_as(_components) mesh_input = torch.from_numpy(mesh_input).type_as(_components)
y_pred = pl_module.predict(mesh_input) y_pred = pl_module.predict(mesh_input)