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

tfl.premade_lib.build_output_calibration_layer

View source on GitHub

Creates a monotonic output calibration layer with inputs range [0, 1].

tfl.premade_lib.build_output_calibration_layer(
    output_calibration_input, model_config, dtype
)

Args:

  • output_calibration_input: Input to the output calibration layer.
  • model_config: Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
  • dtype: dtype

Returns:

A tfl.layers.PWLCalibration instance.