DataReal.get_activation_outhead
- DataReal.get_activation_outhead() str | torch.nn.Module | None [source]
Tries to interfere the activation function for the output head, based on the last transformation. It follows the following rules:
If the last transformation is a MinMaxScaler with a target range of (0, 1), the activation function is sigmoid.
If the last transformation is a MinMaxScaler with a target range of (-1, 1), the activation function is tanh.
Otherwise, no activation function is inferred. None is returned.
- Returns:
Optional[Union[str, nn.Module]] – Activation function for the output head.