tabensemb.model.AbstractModel._check_params#
method
- AbstractModel._check_params(model_name, **kwargs)[source]#
Check the validity of hyperparameters. This is implemented originally for batch_size because TabNet crashes when batch_size is small under certain situations.
- Parameters:
- model_name
The name of a selected model.
- kwargs
Parameters to generate the model. It contains all arguments in
_initial_values().
- Returns:
- dict
The checked kwargs.