tabensemb.model.CatEmbed._check_params

method

CatEmbed._check_params(model_name, **kwargs)

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.