tfl.premade_lib.build_calibration_layers

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

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

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

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