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.
|