tomodrgn.models.mlp_ascii#
- mlp_ascii(input_dim: int, hidden_dims: list[int], output_dim: int)[source]#
Create ASCII art of a fully connected multi-layer perceptron. Constraints: number of digits in each dimension cannot exceed 5. Sample usage:
print_mlp_ascii(input_dim=12345, hidden_dims=[128, 128, 128], output_dim=2)- Parameters:
input_dim – dimensionality of input layer
hidden_dims – list of dimensionalities of hidden layers
output_dim – dimensionality of output layer
- Returns:
None