work on kernel plotting

This commit is contained in:
James Hensman 2014-11-04 09:44:57 +00:00
parent a5d1d3bc03
commit b3aa8fa1e2

View file

@ -161,7 +161,8 @@ def plot(kernel,x=None, fignum=None, ax=None, title=None, plot_limits=None, reso
Xnew = np.vstack((xx.flatten(), yy.flatten())).T
Kx = kernel.K(Xnew, x)
Kx = Kx.reshape(resolution, resolution).T
ax.contour(xx, xx, Kx, vmin=Kx.min(), vmax=Kx.max(), cmap=pb.cm.jet, **mpl_kwargs) # @UndefinedVariable
stop
ax.contour(xx, yy, Kx, vmin=Kx.min(), vmax=Kx.max(), cmap=pb.cm.jet, **mpl_kwargs) # @UndefinedVariable
ax.set_xlim(xmin[0], xmax[0])
ax.set_ylim(xmin[1], xmax[1])
ax.set_xlabel("x1")