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

Add border argument in visualization callback

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-04-29 22:36:10 +02:00
parent e44516fc49
commit db7bb7619f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
prototorch/models/callbacks/visualization.py
View File

@ -267,6 +267,7 @@ class Vis2DAbstract(pl.Callback):
y_train,
title="Prototype Visualization",
cmap="viridis",
border=1,
show_last_only=False,
block=False):
super().__init__()
@ -275,6 +276,7 @@ class Vis2DAbstract(pl.Callback):
self.title = title
self.fig = plt.figure(self.title)
self.cmap = cmap
self.border = border
self.show_last_only = show_last_only
self.block = block
@ -343,8 +345,8 @@ class VisSiameseGLVQ2D(Vis2DAbstract):
s=50,
)
x = np.vstack((x_train, protos))
x_min, x_max = x[:, 0].min() - 1, x[:, 0].max() + 1
y_min, y_max = x[:, 1].min() - 1, x[:, 1].max() + 1
x_min, x_max = x[:, 0].min() - self.border, x[:, 0].max() + self.border
y_min, y_max = x[:, 1].min() - self.border, x[:, 1].max() + self.border
xx, yy = np.meshgrid(np.arange(x_min, x_max, 1 / 50),
np.arange(y_min, y_max, 1 / 50))
mesh_input = np.c_[xx.ravel(), yy.ravel()]