tfl.visualization.draw_model_graph

Draws the model graph.

tfl.visualization.draw_model_graph(
    model_graph,
    calibrator_dpi=30,
    calibrator_figsize=None,
    image_format='png'
)

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)

model_graph a model_info.ModelInfo objects to plot.
calibrator_dpi The DPI for calibrator plots inside the graph nodes.
calibrator_figsize The figsize parameter for calibrator plots.
image_format Format of the image to produce. Using 'svg' format can help with font rendering issues.