Partager via

ModelParameters Classe

  • Référence

Définit des noms de paramètres spécifiques à certains modèles.

Par exemple, pour indiquer les fonctionnalités du jeu de données qui sont catégoriques, un modèle LightGBM accepte le paramètre « categorical_feature » tandis qu’un modèle CatBoost accepte le paramètre « cat_features ».

Héritage
builtins.object
ModelParameters

Constructeur

ModelParameters()

Attributs

CATEGORICAL_FEATURES 

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