tfl.premade_lib.build_linear_combination_layer

Creates a tfl.layers.Linear layer initialized to be an average.

View aliases

Main aliases

tfl.premade.premade_lib.build_linear_combination_layer

tfl.premade_lib.build_linear_combination_layer(
    ensemble_outputs, model_config, dtype
)

Args
ensemble_outputs	Ensemble outputs to be linearly combined.
model_config	Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
dtype	dtype

Returns
A tfl.layers.Linear instance.