tfl.premade_lib.build_calibration_layers

Creates a calibration layer for submodels as list of list of features.

View aliases

Main aliases

tfl.premade.premade_lib.build_calibration_layers

tfl.premade_lib.build_calibration_layers(
    calibration_input_layer,
    model_config,
    layer_output_range,
    submodels,
    separate_calibrators,
    dtype
)

Args
calibration_input_layer	A mapping from feature name to keras.Input.
model_config	Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
layer_output_range	A tfl.premade_lib.LayerOutputRange enum.
submodels	A list of list of feature names.
separate_calibrators	If features should be separately calibrated for each lattice in an ensemble.
dtype	dtype

Returns
A list of list of Tensors representing a calibration layer for submodels.