tabensemb.model.AbstractNN.get_output_norm#

method

static AbstractNN.get_output_norm(task) → Module[source]#

The operation on the output of forward in training/validation/testing steps. This will not affect the input of loss functions.

Parameters:
task

“regression”, “multiclass”, or “binary”

Returns:
nn.Module

The operation on the output.