Attend the Women in ML Symposium on December 7 Register now


Stay organized with collections Save and categorize content based on your preferences.

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.