tfl.premade_lib.set_crystals_lattice_ensemble

View source on GitHub

Extracts crystals from a prefitting model and finalizes model_config.

Used in the notebooks

Used in the tutorials

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.