tfl.premade_lib.set_random_lattice_ensemble

Sets random lattice ensemble in the given model_config.

tfl.premade_lib.set_random_lattice_ensemble(
    model_config, feature_names=None
)

Used in the notebooks

Used in the tutorials

model_config Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
feature_names A list of feature names. If not provided, feature names will be extracted from the feature configs contained in the model_config.