Module model_viz 

Model architecture visualization utilities Model architecture visualization utilities

This module provides utilities for visualizing the architecture of neural network models.

Structs

ModelVizOptions

Options for model architecture visualization

Functions

sequential_model_dataflow

Creates an ASCII representation of the data flow through a sequential model This visualization shows how data flows through the network layers, including transformations in shape and any connections between layers.

sequential_model_summary

Create an ASCII text representation of a sequential model architecture