spamosaic.architectures.wlgcn.WLGCN

class spamosaic.architectures.wlgcn.HEAD(*args: Any, **kwargs: Any)

MLP head that outputs an L2-normalized embedding and reconstructs input features.

Parameters:

• input_size (int) – Input feature dimension.

• output_size (int) – Embedding dimension.

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

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

• dropout (float, default=0.2) – Dropout rate.

• slope (float, default=0.2) – Negative slope for LeakyReLU.