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

tfl.visualization.plot_calibrator_nodes

View source on GitHub

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

tfl.visualization.plot_calibrator_nodes(
    nodes, plot_submodel_calibration=True, font_size=12, axis_label_font_size=14,
    figsize=None
)

Args:

  • nodes: List of calibrator nodes to be plotted.
  • plot_submodel_calibration: If submodel calibrators should be included in the output plot, when more than one calibration node is provided. These are individual calibration layers for each lattice in a lattice ensemble constructed from configs.CalibratedLatticeEnsembleConfig.
  • font_size: Font size for values and labels on the plot.
  • axis_label_font_size: Font size for axis labels.
  • figsize: The figsize parameter passed to pyplot.figure().

Returns:

Pyplot figure object containing the visualisation.