tabensemb.model.PytorchTabular._custom_training_params

method

PytorchTabular._custom_training_params(model_name) → Dict

Customized training settings to override settings in the configuration. The configuration will be restored after training the model.

Parameters:

model_name

The name of a selected model

Returns:

dict

A dict of training parameters.