CatBoostRegressor 类

CatBoost 回归量的模型包装器。

继承

sklearn.base.RegressorMixin

CatBoostRegressor

azureml.automl.runtime.shared.model_wrappers._AbstractModelWrapper

CatBoostRegressor

构造函数

CatBoostRegressor(random_state=0, thread_count=1, **kwargs)

参数

random_state

默认值: 0

thread_count

默认值: 1

方法

fit	拟合 CatBoostRegressor 模型的函数。
get_model	返回 CatBoostRegressor 模型。 否则返回 None。
get_params	CatBoostRegressor 模型的返回参数。
predict	根据数据集特征预测目标。

fit

拟合 CatBoostRegressor 模型的函数。

fit(X, y, **kwargs)

参数

X

必需

输入数据。

y

必需

输入目标值。

kwargs

必需

若是其他参数，请查看 https://catboost.ai/docs/concepts/python-reference_parameters-list.html 获取更多参数。

返回

拟合模型后执行 Self 操作。

get_model

返回 CatBoostRegressor 模型。

否则返回 None。

get_model()

返回

如果已调用拟合方法，则返回已拟合的模型。

get_params

CatBoostRegressor 模型的返回参数。

get_params(deep=True)

参数

deep

默认值: True

如果为 True，也返回子估算器的模型参数。

返回

CatBoostRegressor 模型的参数。

predict

根据数据集特征预测目标。

predict(X)

参数

X

必需

输入数据。

返回

模型预测。