vae_xprize

Classes

`SamplingLayer`
`VAE`

Functions

`_get_encoder`(input_shape, latent_dim)
`_get_decoder`(output_shape, latent_dim)

Package Contents

class vae_xprize.SamplingLayer

Bases: tensorflow.keras.layers.Layer

call(inputs)

vae_xprize._get_encoder(input_shape, latent_dim)

vae_xprize._get_decoder(output_shape, latent_dim)

class vae_xprize.VAE(input_shape, latent_dim, beta_kl=1, beta_reconst=1, encoder=None, decoder=None, **kwargs)

Bases: tensorflow.keras.Model

encoder

decoder

beta_kl

beta_reconst

total_loss_tracker

reconstruction_loss_tracker

kl_loss_tracker

property metrics

call(inputs, training=None, mask=None)

train_step(data)

test_step(data)