tfl.premade_lib.build_output_calibration_layer

View source on GitHub

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

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

A tfl.layers.PWLCalibration instance.