Attend the Women in ML Symposium on December 7 Register now

tfl.visualization.draw_model_graph

Stay organized with collections Save and categorize content based on your preferences.

Draws the model graph.

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.