spamosaic.architectures.hg_lgcn.HG_LGCN

class spamosaic.architectures.hg_lgcn.HG_LGCN(*args: Any, **kwargs: Any)[source]

Heterogeneous Graph LightGCN with encoder–decoder architecture.

Uses HG_LGCN_vanilla as the feature encoder and an MLP/linear decoder for reconstruction.

Parameters:

  • input_size (int) – Input feature dimensionality per node.

  • output_size (int) – Latent embedding dimensionality.

  • K (int, default=8) – Number of LightGCN layers.

  • dec_l (int, default=1) – Number of decoder layers (1 = linear decoder).

  • hidden_size (int, default=512) – Hidden size of the MLP head.

  • dropout (float, default=0.2) – Dropout rate in the MLP head.