DataInt.get_ml_heads
- DataInt.get_ml_heads(head_layer_widths: List[int], last_core_layer_width: int, activation: torch.nn.Module, **kwargs) Tuple[torch.nn.Module, torch.nn.Module] [source]
Returns a fully-connected head for encoding and decoding this feature
- Parameters:
head_layer_widths (List[int]) – List of integers where the length denotes number of layers and values the width of the layers in the head
last_core_layer_width (int) – Dimensionality of autoencoder output to be decoded by this feature head
activation (nn.Module) – Activation function to be used in this head
- Returns:
Tuple[nn.Module, nn.Module] – Tuple containing the input and output head.