tfl.premade_lib.construct_prefitting_model_config

View source on GitHub

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

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.