ncem.models.ModelEDncem

class ncem.models.ModelEDncem(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, cond_type: str = 'gcn', cond_depth: int = 1, cond_dim: int = 8, cond_dropout_rate: float = 0.1, cond_activation: Union[str, keras.engine.base_layer.Layer] = 'relu', cond_l2_reg: float = 0.0, cond_use_bias: bool = True, use_domain: bool = False, use_type_cond: bool = False, scale_node_size: bool = False, output_layer: str = 'gaussian', **kwargs)

Model class for NCEM encoder-decoder with graph layer IND (MAX) or GCN.

Methods