Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

tfl.visualization.plot_all_calibrators

Plots all feature calibrator(s) extracted from a TFL canned estimator.

tfl.visualization.plot_all_calibrators(
    model_graph, num_cols=4, image_format='png', **kwargs
)

The generated plots are arranged in a grid. This function requires IPython and colabtools packages.

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

model_graph a model_info.ModelGraph objects to plot.
num_cols Number of columns in the grid view.
image_format Format of the image to produce.
**kwargs args passed to plot_feature_calibrator and plot_calibrator_nodes.