tfl.premade_lib.set_crystals_lattice_ensemble

Extracts crystals from a prefitting model and finalizes model_config.

Main aliases

tfl.premade.premade_lib.set_crystals_lattice_ensemble

tfl.premade_lib.set_crystals_lattice_ensemble(
    model_config, prefitting_model_config, prefitting_model, feature_names=None
)

Used in the tutorials
TF Lattice Premade Models

Args
`model_config`	Model configuration object describing model architecture. Should be a `tfl.configs.CalibratedLatticeEnsemble` instance.
`prefitting_model_config`	Model configuration object describing prefitting model architecture. Should be a `tfl.configs.CalibratedLatticeEnsemble` insance constructed using `tfl.premade_lib.construct_prefitting_model_config`.
`prefitting_model`	A trained `tfl.premade.CalibratedLatticeEnsemble`, `tfl.estimators.CannedEstimator`, `tfl.estimators.CannedClassifier`, or `tfl.estiamtors.CannedRegressor` instance.
`feature_names`	A list of feature names. If not provided, feature names will be extracted from the feature configs contained in the model_config.

Args