tf.keras.utils.plot_model

View source on GitHub

Converts a Keras model to dot format and save to a file.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.utils.plot_model, `tf.compat.v2.keras.utils.plot_model`

tf.keras.utils.plot_model(
    model, to_file='model.png', show_shapes=False, show_layer_names=True,
    rankdir='TB', expand_nested=False, dpi=96
)

Arguments
model	A Keras model instance
to_file	File name of the plot image.
show_shapes	whether to display shape information.
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 models into clusters.
dpi	Dots per inch.

Returns
A Jupyter notebook Image object if Jupyter is installed. This enables in-line display of the model plots in notebooks.