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

fix: plotting also allows 3D (capitals)

Browse source
This commit is contained in:
mzwiessele 2017-01-09 10:59:51 +00:00
parent 14046705cf
commit fe3bad7200
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
GPy/plotting/gpy_plot/gp_plots.py
View file

@ -81,6 +81,7 @@ def _plot_mean(self, canvas, helper_data, helper_prediction,
**kwargs):
_, free_dims, Xgrid, x, y, _, _, resolution = helper_data
plots = dict()
if len(free_dims)<=2:
mu, _, _ = helper_prediction
if len(free_dims)==1:
@ -88,12 +89,12 @@ def _plot_mean(self, canvas, helper_data, helper_prediction,
update_not_existing_kwargs(kwargs, pl().defaults.meanplot_1d) # @UndefinedVariable
plots = dict(gpmean=[pl().plot(canvas, Xgrid[:, free_dims], mu, label=label, **kwargs)])
else:
if projection == '2d':
if projection.lower() in '2d':
update_not_existing_kwargs(kwargs, pl().defaults.meanplot_2d) # @UndefinedVariable
plots = dict(gpmean=[pl().contour(canvas, x[:,0], y[0,:],
mu.reshape(resolution, resolution).T,
levels=levels, label=label, **kwargs)])
elif projection == '3d':
elif projection.lower() in '3d':
update_not_existing_kwargs(kwargs, pl().defaults.meanplot_3d) # @UndefinedVariable
plots = dict(gpmean=[pl().surface(canvas, x, y,
mu.reshape(resolution, resolution),