tirank.Loss.MSE_loss

tirank.Loss.MSE_loss(y_pred, y_true)[source]

Calculates the Mean Squared Error (MSE) loss for regression.

Wrapper for nn.MSELoss.

Parameters:

y_pred (torch.Tensor) – Predicted values.
y_true (torch.Tensor) – True values.

Returns:

The scalar MSE loss.

Return type:

torch.Tensor