ncem.models.ModelED

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

Model class for non-spatial encoder-decoder.

Methods

Parameters
  • latent_dim (int) –

  • dropout_rate (float) –

  • l2_coef (float) –

  • l1_coef (float) –

  • enc_intermediate_dim (int) –

  • enc_depth (int) –

  • dec_intermediate_dim (int) –

  • dec_depth (int) –

  • use_domain (bool) –

  • use_type_cond (bool) –

  • scale_node_size (bool) –

  • output_layer (str) –