Compartilhar via


ModelParameters Classe

Define nomes de parâmetro específicos para determinados modelos.

Por exemplo, para indicar quais recursos no conjuntos de dados são categóricos, um modelo LightGBM aceita o parâmetro 'categorical_feature', enquanto um modelo CatBoost aceita o parâmetro 'cat_features'.

Herança
builtins.object
ModelParameters

Construtor

ModelParameters()

Atributos

CATEGORICAL_FEATURES

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