tirank.Model.DenseNetEncoderModel

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

Bases: Module

DenseNet-style encoder network.

Parameters:

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

  • nlayers (int) – Number of dense layers.

  • n_output (int) – Output embedding dimension.

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

  • growth_rate (float, optional) – Growth rate for the dense layers. Defaults to 0.5.

forward(x)[source]

Forward pass for the DenseNet encoder.

Parameters:

x (torch.Tensor) – Input feature tensor.

Returns:

Output embedding tensor.

Return type:

torch.Tensor