tirank.Model.MLPEncoderModel

class tirank.Model.MLPEncoderModel(*args: Any, **kwargs: Any)[source]

Bases: Module

MLP-based (Multi-Layer Perceptron) encoder network.

Parameters:

  • n_features (int) – Input feature size (number of gene pairs).

  • nhid (int) – Dimension of the hidden layers.

  • nlayers (int) – Total number of layers (input, hidden, output).

  • n_output (int) – Output embedding dimension.

  • dropout (float, optional) – Dropout value. Defaults to 0.5.

forward(x)[source]

Forward pass for the MLP encoder.

Parameters:

x (torch.Tensor) – Input feature tensor.

Returns:

Output embedding tensor.

Return type:

torch.Tensor