![]() |
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 |
---|
Args | |
---|---|
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. |
Returns | |
---|---|
A tfl.configs.CalibratedLatticeEnsembleConfig instance.
|