ncem.models.ModelCVAE

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

Model class for conditional variational autoencoder.

Methods