tf.keras.utils.model_to_dot

View source on GitHub

Convert a Keras model to dot format.

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/utils/model_to_dot