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=96, subgraph=False
)
Args |
|---|
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 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.
|
TensorFlow 1 version
View source on GitHub