mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-03 16:52:39 +02:00
fix: plotting
This commit is contained in:
parent
93a7c08892
commit
4f3047e035
2 changed files with 6 additions and 3 deletions
|
|
@ -268,7 +268,10 @@ def _plot_density(self, canvas, helper_data, helper_prediction, label, **kwargs)
|
|||
# 1D plotting:
|
||||
fills = []
|
||||
for d in range(mu.shape[1]):
|
||||
fills.append(pl().fill_gradient(canvas, Xgrid[:, free_dims[0]], [p[:,d] for p in percs], label=label, **kwargs))
|
||||
fills.append(pl().fill_gradient(
|
||||
canvas, Xgrid[:, free_dims[0]], [p[:,d] for p in percs],
|
||||
label=label, **kwargs)
|
||||
)
|
||||
return dict(gpdensity=fills)
|
||||
else:
|
||||
pass # Nothing to plot!
|
||||
|
|
|
|||
|
|
@ -90,7 +90,7 @@ class MatplotlibPlots(AbstractPlottingLibrary):
|
|||
#ax.legend(prop=fontdict)
|
||||
legend_ontop(ax, ncol=legend, fontdict=fontdict)
|
||||
if title is not None: ax.figure.suptitle(title)
|
||||
return ax
|
||||
return plots
|
||||
|
||||
def show_canvas(self, ax, **kwargs):
|
||||
ax.figure.canvas.draw()
|
||||
|
|
@ -298,7 +298,7 @@ class MatplotlibPlots(AbstractPlottingLibrary):
|
|||
from matplotlib.collections import PolyCollection
|
||||
if 'zorder' not in kwargs:
|
||||
kwargs['zorder'] = 0
|
||||
plots.append(PolyCollection(polycol, **kwargs))
|
||||
plots.append(PolyCollection(polycol, label=label, **kwargs))
|
||||
ax.add_collection(plots[-1], autolim=True)
|
||||
ax.autoscale_view()
|
||||
return plots
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue