tfl.premade_lib.construct_prefitting_model_config

Constructs a model config for a prefitting model for crystal extraction.

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

Used in the notebooks

Used in the tutorials

model_config Model configuration object describing model architecture. Should be a tfl.configs.CalibratedLatticeEnsemble 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.

A tfl.configs.CalibratedLatticeEnsembleConfig instance.