Model plotting utilities

[source]

plot_model function

keras.utils.plot_model( model, to_file="model.png", show_shapes=False, show_dtype=False, show_layer_names=False, rankdir="TB", expand_nested=False, dpi=200, show_layer_activations=False, show_trainable=False, **kwargs ) 

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

Example

inputs = ... outputs = ... model = keras.Model(inputs=inputs, outputs=outputs) dot_img_file = '/tmp/model_1.png' keras.utils.plot_model(model, to_file=dot_img_file, show_shapes=True) 

Arguments

• model: A Keras model instance
• to_file: File name of the plot image.
• 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.
• show_layer_activations: Display layer activations (only for layers that have an activation property).
• show_trainable: whether to display if a layer is trainable.

Returns

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

[source]

model_to_dot function

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 ) 

Convert a Keras model to dot format.

Arguments

• 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.

Returns

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