ModelParameters Class

Defines parameter names specific to certain models.

For example, to indicate which features in the dataset are categorical a LightGBM model accepts the 'categorical_feature' parameter while a CatBoost model accepts the 'cat_features' parameter.

Constructor

ModelParameters()

Attributes

CATEGORICAL_FEATURES

CATEGORICAL_FEATURES = {'CatBoostClassifier': 'cat_features', 'CatBoostRegressor': 'cat_features', 'LightGBMClassifier': 'categorical_feature', 'LightGBMRegressor': 'categorical_feature', 'TabnetClassifier': 'categorical_features', 'TabnetRegressor': 'categorical_features'}