evalml.pipelines.components.CatBoostRegressor

Inheritance diagram of CatBoostRegressor
class evalml.pipelines.components.CatBoostRegressor(n_estimators=1000, eta=0.03, max_depth=6, bootstrap_type=None, random_state=0)[source]

CatBoost Regressor, a regressor that uses gradient-boosting on decision trees. CatBoost is an open-source library and natively supports categorical features.

For more information, check out https://catboost.ai/

name = 'CatBoost Regressor'
model_family = 'catboost'
supported_problem_types = [<ProblemTypes.REGRESSION: 'regression'>]
hyperparameter_ranges = {'eta': Real(low=0, high=1, prior='uniform', transform='identity'), 'max_depth': Integer(low=1, high=16, prior='uniform', transform='identity'), 'n_estimators': Integer(low=10, high=1000, prior='uniform', transform='identity')}

Instance attributes

SEED_MAX

SEED_MIN

feature_importances

Returns feature importances.

Methods:

__init__

Initialize self.

describe

Describe a component and its parameters

fit

Build a model

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.