GitHub

tfl.visualization.draw_model_graph

Draws the model graph.

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

Used in the tutorials
TF Lattice Canned Estimators Shape Constraints with Tensorflow Lattice

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2021-02-19 UTC.

Args