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) –