TensorFlow 2 version
View source on GitHub
Convert a Keras model to dot format.
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.compat.v1.keras.utils.model_to_dot, `tf.compat.v2.keras.utils.model_to_dot`
tf.keras.utils.model_to_dot(
model, show_shapes=False, show_layer_names=True, rankdir='TB',
expand_nested=False, dpi=96, subgraph=False
)
Arguments | |
|---|---|
model
|
A Keras model instance. |
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. |
subgraph
|
whether to return a pydot.Cluster instance.
|
Returns | |
|---|---|
A pydot.Dot instance representing the Keras model or
a pydot.Cluster instance representing nested model if
subgraph=True.
|
Raises | |
|---|---|
ImportError
|
if graphviz or pydot are not available. |