CatBoostClassifier 类

CatBoost 分类器的模型包装器。

继承

sklearn.base.ClassifierMixin

CatBoostClassifier

azureml.automl.runtime.shared.model_wrappers._AbstractModelWrapper

CatBoostClassifier

构造函数

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

参数

random_state

默认值: 0

thread_count

默认值: 1

方法

fit	CatBoostClassifier 模型的拟合函数。
get_model	返回 CatBoostClassifier 模型。 否则返回 None。
get_params	返回 CatBoostClassifier 模型的参数。
predict	根据数据集特征预测目标。
predict_proba	根据数据集特征预测每个类的概率。

fit

CatBoostClassifier 模型的拟合函数。

fit(X, y, **kwargs)

参数

X

必需

输入数据。

y

必需

输入目标值。

kwargs

必需

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

返回

拟合模型后执行 Self 操作。

get_model

返回 CatBoostClassifier 模型。

否则返回 None。

get_model()

返回

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

get_params

返回 CatBoostClassifier 模型的参数。

get_params(deep=True)

参数

deep

默认值: True

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

返回

CatBoostClassifier 模型的参数。

predict

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

predict(X)

参数

X

必需

输入数据。

返回

模型预测。

predict_proba

根据数据集特征预测每个类的概率。

predict_proba(X)

参数

X

必需

输入数据。

返回

模型预测的每个类的概率。