tabensemb.model.WideDeep._update_optimizer_lr_scheduler_params

method

WideDeep._update_optimizer_lr_scheduler_params(model_name, **kwargs) → Tuple[str, Dict, str, Dict]

Update parameters of the optimizer and the lr_scheduler according to the input hyperparameters when initializing a model.

Parameters:

model_name

The name of the model

kwargs

Parameters to train the model returned by _get_params(). It contains all arguments in _initial_values().

Returns:

str

The name of the optimizer in torch.optim

Dict

The parameters of the optimizer

str

The name of the lr scheduler in torch.optim.lr_scheduler

Dict

The parameters of the lr scheduler