spamosaic.architectures.wlgcn.WLGCN

class spamosaic.architectures.wlgcn.WLGCN(*args: Any, **kwargs: Any)[source]

Deep WLGCN with encoder-decoder structure for representation learning.

Parameters:

  • input_size (int) – Input feature dimension.

  • output_size (int) – Output embedding dimension.

  • K (int, optional) – Number of GCN propagation steps (default: 8).

  • dec_l (int, optional) – Number of layers in the decoder (1 or 2). Default is 1.

  • hidden_size (int, optional) – Hidden layer size for encoder. Default is 512.

  • dropout (float, optional) – Dropout rate. Default is 0.2.

  • slope (float, optional) – LeakyReLU negative slope. Default is 0.2.