tabensemb.model.AbstractNN.output_norm#
method
- AbstractNN.output_norm(y_pred)[source]#
User defined operation before output. This is not related to the input of
loss_fn().- Parameters:
- y_pred
The prediction from the model (from
forward()and afterbefore_loss_fn()).
- Returns:
- torch.Tensor
The modified prediction.