示例#1
0
文件: iwae.py 项目: dementrock/iwae
def get_samples(model, num_samples, seed=123):
    srng = utils.srng(seed)
    prior_samples = model.prior.samplesIshape_srng((num_samples, model.first_p_layer_weights_np().shape[0]), srng)
    samples = [prior_samples]
    for layer in model.p_layers[:-1]:
        samples.append(layer.samplesIx_srng(samples[-1], srng))
    samples_function = theano.function([], model.p_layers[-1].meanIx(samples[-1]))

    return reshape_and_tile_images(samples_function())
示例#2
0
def plot_images(images, shape, path, filename):
     # finally save to file
    import matplotlib
    matplotlib.use('Agg')
    import matplotlib.pyplot as plt
    
    images = utils.reshape_and_tile_images(images, shape)
    plt.imshow(images, cmap='Greys')
    plt.axis('off')
    plt.savefig(path + filename + ".svg", format="svg")
    plt.close()
示例#3
0
文件: iwae.py 项目: shafiahmed/iwae
def get_samples(model, num_samples, seed=123):
    srng = utils.srng(seed)
    prior_samples = model.prior.samplesIshape_srng(
        (num_samples, model.first_p_layer_weights_np().shape[0]), srng)
    samples = [prior_samples]
    for layer in model.p_layers[:-1]:
        samples.append(layer.samplesIx_srng(samples[-1], srng))
    samples_function = theano.function([],
                                       model.p_layers[-1].meanIx(samples[-1]))

    return reshape_and_tile_images(samples_function())
示例#4
0
def get_last_p_layer_weights(model):
    shape = (28, 28)
    return utils.reshape_and_tile_images(model.last_p_layer_weights_np(),
                                         shape=shape)
示例#5
0
def get_first_q_layer_weights(model):
    shape = (28, 28)
    return utils.reshape_and_tile_images(model.first_q_layer_weights_np().T,
                                         shape=shape)
示例#6
0
文件: iwae.py 项目: dementrock/iwae
def get_last_p_layer_weights(model):
    return utils.reshape_and_tile_images(model.last_p_layer_weights_np())
示例#7
0
文件: iwae.py 项目: dementrock/iwae
def get_first_q_layer_weights(model):
    return utils.reshape_and_tile_images(model.first_q_layer_weights_np().T)
示例#8
0
文件: iwae.py 项目: shafiahmed/iwae
def get_last_p_layer_weights(model):
    return utils.reshape_and_tile_images(model.last_p_layer_weights_np())
示例#9
0
文件: iwae.py 项目: shafiahmed/iwae
def get_first_q_layer_weights(model):
    return utils.reshape_and_tile_images(model.first_q_layer_weights_np().T)