better bound extimate for plot_latent background

This commit is contained in:
Max Zwiessele 2013-08-02 16:39:34 +01:00
parent d66a8e02a5
commit 56b86542d4

View file

@ -46,8 +46,9 @@ def plot_latent(model, labels=None, which_indices=None,
Xtest_full[:, [input_1, input_2]] = x
mu, var, low, up = model.predict(Xtest_full)
var = var[:, :1]
return var
view = ImshowController(ax, plot_function, tuple(xmin) + tuple(xmax),
return np.log(var)
view = ImshowController(ax, plot_function,
tuple(model.X.min(0)[:, [input_1, input_2]]) + tuple(model.X.max(0)[:, [input_1, input_2]]),
resolution, aspect=aspect, interpolation='bilinear',
cmap=pb.cm.binary)
@ -124,7 +125,8 @@ def plot_magnification(model, labels=None, which_indices=None,
Xtest_full[:, [input_1, input_2]] = x
mf=model.magnification(Xtest_full)
return mf
view = ImshowController(ax, plot_function, tuple(xmin) + tuple(xmax),
view = ImshowController(ax, plot_function,
tuple(model.X.min(0)[:, [input_1, input_2]]) + tuple(model.X.max(0)[:, [input_1, input_2]]),
resolution, aspect=aspect, interpolation='bilinear',
cmap=pb.cm.gray)