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

plotting behaviour adapted for BGPLVM

Browse source
This commit is contained in:
Max Zwiessele 2013-06-04 18:26:16 +01:00
parent 2b0858b697
commit 8bdb14b0f9
1 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
GPy/models/mrd.py
View file

@ -256,19 +256,19 @@ class MRD(model):
self.Z = Z
return Z
def _handle_plotting(self, fignum, ax, plotf):
if ax is None:
def _handle_plotting(self, fignum, axes, plotf):
if axes is None:
fig = pylab.figure(num=fignum, figsize=(4 * len(self.bgplvms), 3))
for i, g in enumerate(self.bgplvms):
if ax is None:
ax = fig.add_subplot(1, len(self.bgplvms), i + 1)
elif isinstance(ax, (tuple, list)):
ax = ax[i]
if axes is None:
axes = fig.add_subplot(1, len(self.bgplvms), i + 1)
elif isinstance(axes, (tuple, list)):
axes = axes[i]
else:
raise ValueError("Need one ax per latent dimension Q")
plotf(i, g, ax)
raise ValueError("Need one axes per latent dimension Q")
plotf(i, g, axes)
pylab.draw()
if ax is None:
if axes is None:
fig.tight_layout()
return fig
else:
@ -286,11 +286,11 @@ class MRD(model):
return fig
def plot_scales(self, fignum="MRD Scales", ax=None, *args, **kwargs):
fig = self._handle_plotting(fignum, ax, lambda i, g, ax: g.kern.plot_ARD(ax=ax, *args, **kwargs))
fig = self._handle_plotting(fignum, ax, lambda i, g, ax: g.kern.plot_ARD(axes=ax, *args, **kwargs))
return fig
def plot_latent(self, fignum="MRD Latent Spaces", ax=None, *args, **kwargs):
fig = self._handle_plotting(fignum, ax, lambda i, g, ax: g.plot_latent(ax=ax, *args, **kwargs))
fig = self._handle_plotting(fignum, ax, lambda i, g, ax: g.plot_latent(axes=ax, *args, **kwargs))
return fig
def _debug_plot(self):