tabensemb.model.AbstractNN.before_loss_fn

method

AbstractNN.before_loss_fn(y, yhat)

Treatments on the prediction and the ground truth before passing them to loss_fn().

Parameters:

y

The prediction from forward().

yhat

The ground truth.

Returns:

torch.Tensor

The processed prediction

torch.Tensor

The processed ground truth