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][source]

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.