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=96, subgraph=False
)
| 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 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. |
