TensorFlow 1 version View source on GitHub

Converts a Keras model to dot format and save to a file.

    model, to_file='model.png', show_shapes=False, show_layer_names=True,
    rankdir='TB', expand_nested=False, dpi=96

Used in the notebooks

Used in the guide Used in the tutorials


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


A Jupyter notebook Image object if Jupyter is installed. This enables in-line display of the model plots in notebooks.