tabensemb.model.PytorchTabular.required_models

method

PytorchTabular.required_models(model_name: str) → List[str] | None

The names of models required by the requested model. If not None and the required model is trained, the required model will be passed to _new_model(). If models from other model bases are required, the name should be EXTERN_{Name of the model base}_{Name of the model}

Notes

For TorchModel, if the required model is in the TorchModel itself, the AbstractNN is passed to _new_model(); if the required model is in another model base, the AbstractModel is passed.