View source on GitHub |
Creates a tfl.layers.Linear
layer initialized to be an average.
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.
|