tirank.Model.RegscorePredictor

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

Bases: Module

Prediction head for ‘Regression’ analysis.

Predicts a single continuous value. No output activation is applied.

Parameters:

  • n_features (int) – Input embedding dimension (from encoder).

  • nhid (int) – Hidden dimension of the predictor MLP.

  • nhout (int, optional) – Output dimension. Defaults to 1.

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

forward(embedding)[source]

Forward pass for the regression score predictor.

Parameters:

embedding (torch.Tensor) – Input embedding tensor from the encoder.

Returns:

Predicted continuous value (scalar tensor).

Return type:

torch.Tensor