def plot(self, V, Wf, Wt, Z, H, curriter=-1): rank = len(Z) nrows = rank + 2 WZH = self.reconstruct(Wf, Wt, Z, H, circular=self.circular) W = Basilica.w_product(Wf, Wt) plottools.plotall([V, WZH] + [SIPLCA2.reconstruct(W[:,z,:], Z[z], H[z,:], circular=self.circular) for z in xrange(len(Z))], title=['V (Iteration %d)' % curriter, 'Reconstruction'] + ['Basis %d reconstruction' % x for x in xrange(len(Z))], colorbar=False, grid=False, cmap=plt.cm.hot, subplot=(nrows, 2), order='c', align='xy') plottools.plotall([None] + [Z], subplot=(nrows, 2), clf=False, plotfun=lambda x: plt.bar(np.arange(len(x)) - 0.4, x), xticks=[[], range(rank)], grid=False, colorbar=False, title='Z') plots = [None] * (3*nrows + 2) titles = plots + ['W%d' % x for x in range(rank)] wxticks = [[]] * (3*nrows + rank + 1) + [range(0, W.shape[2], 10)] plots.extend(W.transpose((1, 0, 2))) plottools.plotall(plots, subplot=(nrows, 6), clf=False, order='c', align='xy', cmap=plt.cm.hot, colorbar=False, ylabel=r'$\parallel$', grid=False, title=titles, yticks=[[]], xticks=wxticks) plots = [None] * (2*nrows + 2) titles=plots + ['H%d' % x for x in range(rank)] if np.squeeze(H).ndim < 4: plotH = np.squeeze(H) else: plotH = H.sum(2) if rank == 1: plotH = [plotH] plots.extend(plotH) plottools.plotall(plots, subplot=(nrows, 3), order='c', align='xy', grid=False, clf=False, title=titles, yticks=[[]], colorbar=False, cmap=plt.cm.hot, ylabel=r'$*$', xticks=[[]]*(3*nrows-1) + [range(0, V.shape[1], 100)]) plt.draw()
def reconstruct(Wf, Wt, Z, H, norm=1.0, circular=False): W = Basilica.w_product(Wf, Wt) return SIPLCA2.reconstruct(W, Z, H, norm, circular)