ncem.models.ModelED

class ncem.models.ModelED(input_shapes, latent_dim: int = 10, dropout_rate: float = 0.1, l2_coef: float = 0.0, l1_coef: float = 0.0, enc_intermediate_dim: int = 128, enc_depth: int = 2, dec_intermediate_dim: int = 128, dec_depth: int = 2, use_domain: bool = False, use_type_cond: bool = True, scale_node_size: bool = False, output_layer: str = 'gaussian', **kwargs)

Model class for non-spatial encoder-decoder.

Methods