torchcell.trainers.neo_regression.MSEListMLELoss

class MSEListMLELoss(alpha: float = 0.5)

Bases: Module

Loss combining MSE with an alpha-weighted ListMLE ranking term.

forward(y_pred: Tensor, y_true: Tensor) → Tensor

Return MSE plus alpha times the ListMLE loss.