Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfl.visualization.draw_model_graph

View source on GitHub

Draws the model graph.

tfl.visualization.draw_model_graph(
    model_graph, calibrator_dpi=30
)

Used in the notebooks

Used in the tutorials

This function requires IPython and graphviz packages.

model_graph = estimators.get_model_graph(saved_model_path)
visualization.draw_model_graph(model_graph)

Args:

  • model_graph: a model_info.ModelInfo objects to plot.
  • calibrator_dpi: The DPI for calibrator plots inside the graph nodes.