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.
|