fix: plotting

This commit is contained in:
mzwiessele 2018-09-19 21:52:58 +01:00
parent 93a7c08892
commit 4f3047e035
2 changed files with 6 additions and 3 deletions

View file

@ -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!

View file

@ -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