diagnostics.py 文件源码

def plot(self):
        if self.vis is not None:
            pylab.figure('log probs')
            pylab.clf()
            plot_objfn(self.fe_info['main'], self.log_Z_info['main'], 'b', label='Raw')
            plot_objfn(self.fe_info['avg'], self.log_Z_info['avg'], 'r', label='Averaged')
            pylab.title('log probs')
            pylab.legend(loc='lower right')
            pylab.gcf().canvas.draw()

            pylab.figure('log probs (zoomed)')
            pylab.clf()
            plot_objfn(self.fe_info['main'], self.log_Z_info['main'], 'b', zoom=True, label='Raw')
            plot_objfn(self.fe_info['avg'], self.log_Z_info['avg'], 'r', label='Averaged')
            pylab.title('log probs (zoomed)')
            pylab.legend(loc='lower right')
            pylab.gcf().canvas.draw()

        if self.target_moments is not None:
            pylab.figure('moment matching objective')
            pylab.clf()
            plot_objfn(self.dp_info['main'], self.log_Z_info['main'], 'b', label='Raw')
            plot_objfn(self.dp_info['avg'], self.log_Z_info['avg'], 'r', label='Averaged')
            pylab.title('moment matching objective')
            pylab.legend(loc='lower right')
            pylab.gcf().canvas.draw()

            pylab.figure('moment matching objective (zoomed)')
            pylab.clf()
            plot_objfn(self.dp_info['main'], self.log_Z_info['main'], 'b', zoom=True, label='Raw')
            plot_objfn(self.dp_info['avg'], self.log_Z_info['avg'], 'r', label='Averaged')
            pylab.title('moment matching objective (zoomed)')
            pylab.legend(loc='lower right')
            pylab.gcf().canvas.draw()