rbm_vis.py 文件源码

python
阅读 20 收藏 0 点赞 0 评论 0

项目:fang 作者: rgrosse 项目源码 文件源码
def show_particles(rbm, state, dataset, display=True, figname='PCD particles', figtitle='PCD particles',
                   size=None):
    try:
        fantasy_vis = rbm.vis_expectations(state.h)
    except:
        fantasy_vis = state

    if size is None:
        size = (dataset.num_rows, dataset.num_cols)
    imgs = [fantasy_vis[j, :np.prod(size)].reshape(size).as_numpy_array()
            for j in range(fantasy_vis.shape[0])]
    visual = misc.norm01(misc.pack(imgs))
    if display:
        pylab.figure(figname)
        pylab.matshow(visual, cmap='gray', fignum=False)
        pylab.title(figtitle)
    return visual
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号