ncem.models.ModelCVAE

class ncem.models.ModelCVAE(input_shapes, latent_dim=10, intermediate_dim_enc=128, intermediate_dim_dec=128, depth_enc=1, depth_dec=1, dropout_rate=0.1, l2_coef=0.0, l1_coef=0.0, use_domain=False, use_type_cond=True, use_batch_norm=False, scale_node_size=False, transform_input=False, output_layer='gaussian', **kwargs)

Model class for conditional variational autoencoder.

Methods

Parameters

• latent_dim (int) –

• intermediate_dim_enc (int) –

• intermediate_dim_dec (int) –

• depth_enc (int) –

• depth_dec (int) –

• dropout_rate (float) –

• l2_coef (float) –

• l1_coef (float) –

• use_domain (bool) –

• use_type_cond (bool) –

• use_batch_norm (bool) –

• scale_node_size (bool) –

• transform_input (bool) –