dgs.utils.nn.set_up_hidden_layer_sizes

dgs.utils.nn.set_up_hidden_layer_sizes(input_size: int, output_size: int, hidden_sizes: list[int] | None = None) → list[int]

Given the input and output size of an FC-NN, create a list of the sizes containing each hidden layer in the network. There might be zero hidden layers.

Params:

input_size: The size of the input to the FC-Layers. output_size: Output-size of the FC-Layers. hidden_layers: The dimensionality of each hidden layer in this network. Default None means no hidden layers.

Returns:

The sizes of the hidden layers including input and output size.