tf.keras.utils.plot_model
View source on GitHub |
Converts a Keras model to dot format and save to a file.
tf.keras.utils.plot_model( model, to_file='model.png', show_shapes=False, show_layer_names=True, rankdir='TB', expand_nested=False, dpi=96 )
Arguments | |
---|---|
model |
A Keras model instance |
to_file |
File name of the plot image. |
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. |
Returns | |
---|---|
A Jupyter notebook Image object if Jupyter is installed. This enables in-line display of the model plots in notebooks. |
© 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/plot_model