poli.objective_repository.gfp_cbas.make_vae.SimpleVAE

class poli.objective_repository.gfp_cbas.make_vae.SimpleVAE(input_shape, latent_dim, flatten=True, *args, **kwargs)

Basic VAE where the encoder and decoder can be constructed from lists of layers

__init__(input_shape, latent_dim, flatten=True, *args, **kwargs)

Methods

__init__(input_shape, latent_dim[, flatten])
add_decoder_layer(layer)	Append a keras Layer to self.decoderLayers_
add_encoder_layer(layer)	Append a keras Layer to self.encoderLayers_
build_decoder(decode_activation)	Construct the decoder from list of layers
build_encoder()	Construct the encoder from list of layers
build_vae([epsilon_std, kl_scale])	Build the VAE
compile(*args, **kwargs)
fit(*args, **kwargs)
load_all_weights(prefix)
plot_model(*args, **kwargs)
save_all_weights(prefix)