ModelParameters 类

定义特定于特定模型的参数名称。

例如,若要指示数据集中的哪些特征是分类的,LightGBM 模型接受“categorical_feature”参数,而 CatBoost 模型接受“cat_features”参数。

构造函数

ModelParameters()

属性

CATEGORICAL_FEATURES

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