tf.keras.utils.model_to_dot

Convert a Keras model to dot format.

tf.keras.utils.model_to_dot(
    model,
    show_shapes=False,
    show_dtype=False,
    show_layer_names=True,
    rankdir='TB',
    expand_nested=False,
    dpi=200,
    subgraph=False,
    show_layer_activations=False,
    show_trainable=False,
    **kwargs
)

model A Keras model instance. show_shapes whether to display shape information. show_dtype whether to display layer dtypes. show_layer_names whether to display layer names. rankdir rankdir argument passed to PyDot, a string specifying the format of the plot: "TB" creates a vertical plot; "LR" creates a horizontal plot. expand_nested whether to expand nested Functional models into clusters. dpi Image resolution in dots per inch. subgraph whether to return a pydot.Cluster instance. show_layer_activations Display layer activations (only for layers that have an activation property). show_trainable whether to display if a layer is trainable.

A pydot.Dot instance representing the Keras model or a pydot.Cluster instance representing nested model if subgraph=True.