tf.keras.utils.plot_model(
model,
to_file='model.png',
show_shapes=False,
show_layer_names=True,
rankdir='TB'
)
Defined in tensorflow/python/keras/utils/vis_utils.py
.
Converts a Keras model to dot format and save to a file.
Arguments:
model
: A Keras model instanceto_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.