apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-08 19:42:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix stick man example

Browse source
This commit is contained in:
Zhenwen Dai 2014-04-09 12:22:46 +01:00
parent 5cfc250ad1
commit 9d12c83935
6 changed files with 15 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

9
GPy/plotting/matplot_dep/visualize.py
View file

@ -85,6 +85,7 @@ class vector_show(matplotlib_show):
class lvm(matplotlib_show):
def __init__(self, vals, model, data_visualize, latent_axes=None, sense_axes=None, latent_index=[0,1]):
"""Visualize a latent variable model
@ -98,7 +99,7 @@ class lvm(matplotlib_show):
vals = param_to_array(model.X.mean)
else:
vals = param_to_array(model.X)
vals = param_to_array(vals)
matplotlib_show.__init__(self, vals, axes=latent_axes)
@ -121,7 +122,7 @@ class lvm(matplotlib_show):
self.move_on = False
self.latent_index = latent_index
self.latent_dim = model.input_dim
# The red cross which shows current latent point.
self.latent_values = vals
self.latent_handle = self.latent_axes.plot([0],[0],'rx',mew=2)[0]
@ -130,10 +131,10 @@ class lvm(matplotlib_show):
def modify(self, vals):
"""When latent values are modified update the latent representation and ulso update the output visualization."""
self.vals = vals.copy()
self.vals = vals[None,:].copy()
y = self.model.predict(self.vals)[0]
self.data_visualize.modify(y)
self.latent_handle.set_data(self.vals[self.latent_index[0]], self.vals[self.latent_index[1]])
self.latent_handle.set_data(self.vals[:,self.latent_index[0]], self.vals[:,self.latent_index[1]])
self.axes.figure.canvas.draw()