tfl.premade_lib.build_calibrated_lattice_ensemble_layer

Creates a calibration layer followed by a lattice ensemble layer.

calibration_input_layer A mapping from feature name to tf.keras.Input.
model_config Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
average_outputs Whether to average the outputs of this layer.
dtype dtype

A tfl.layers.RTL instance if model_config.lattices is 'rtl_layer. Otherwise a list of tfl.layers.Lattice instances.