catboost_classifier#
CatBoost 分类器,一种使用决策树梯度提升的分类器。CatBoost 是一个开源库,原生支持类别特征。
模块内容#
类摘要#
CatBoost 分类器,一种使用决策树梯度提升的分类器。CatBoost 是一个开源库,原生支持类别特征。 |
目录#
- class evalml.pipelines.components.estimators.classifiers.catboost_classifier.CatBoostClassifier(n_estimators=10, eta=0.03, max_depth=6, bootstrap_type=None, silent=True, allow_writing_files=False, random_seed=0, n_jobs=-1, **kwargs)[source]#
CatBoost 分类器,一种使用决策树梯度提升的分类器。CatBoost 是一个开源库,原生支持类别特征。
欲了解更多信息,请访问 https://catboost.ai/
- 参数
n_estimators (float) – 要构建的最大树数量。默认为 10。
eta (float) – 学习率。默认为 0.03。
max_depth (int) – 基学习器的最大树深度。默认为 6。
bootstrap_type (string) – 定义采样对象权重的方法。可用方法为 ‘Bayesian’、‘Bernoulli’、‘MVS’。默认为 None。
silent (boolean) – 是否使用“静默”日志模式。默认为 True。
allow_writing_files (boolean) – 训练时是否允许写入快照文件。默认为 False。
n_jobs (int or None) – 并行运行的作业数量。-1 表示使用所有进程。默认为 -1。
random_seed (int) – 随机数生成器的种子。默认为 0。
属性
hyperparameter_ranges
{ “n_estimators”: Integer(4, 100), “eta”: Real(0.000001, 1), “max_depth”: Integer(4, 10),}
model_family
ModelFamily.CATBOOST
modifies_features
True
modifies_target
False
name
CatBoost 分类器
supported_problem_types
[ ProblemTypes.BINARY, ProblemTypes.MULTICLASS, ProblemTypes.TIME_SERIES_BINARY, ProblemTypes.TIME_SERIES_MULTICLASS,]
training_only
False
方法
使用相同的参数和随机状态构造一个新的组件。
返回此组件的默认参数。
描述一个组件及其参数。
拟合的 CatBoost 分类器的特征重要性。
将 CatBoost 分类器组件拟合到数据。
使用拟合的回归器查找预测区间。
从文件路径加载组件。
返回布尔值,确定组件在调用 predict、predict_proba、transform 或 feature_importances 之前是否需要拟合。
返回用于初始化组件的参数。
使用拟合的 CatBoost 分类器进行预测。
使用拟合的 CatBoost 分类器进行概率预测。
将组件保存到文件路径。
更新组件的参数字典。
- clone(self)#
使用相同的参数和随机状态构造一个新的组件。
- 返回
此组件的一个新实例,具有相同的参数和随机状态。
- default_parameters(cls)#
返回此组件的默认参数。
我们的约定是 Component.default_parameters == Component().parameters。
- 返回
此组件的默认参数。
- 返回类型
dict
- describe(self, print_name=False, return_dict=False)#
描述一个组件及其参数。
- 参数
print_name (bool, optional) – 是否打印组件名称
return_dict (bool, optional) – 是否以字典格式 {“name”: name, “parameters”: parameters} 返回描述
- 返回
如果 return_dict 为 True,则返回字典,否则返回 None。
- 返回类型
None or dict
- property feature_importance(self)#
拟合的 CatBoost 分类器的特征重要性。
- fit(self, X, y=None)[source]#
将 CatBoost 分类器组件拟合到数据。
- 参数
X (pd.DataFrame) – 输入训练数据,形状为 [n_samples, n_features]。
y (pd.Series) – 目标训练数据,长度为 [n_samples]。
- 返回
self
- get_prediction_intervals(self, X: pandas.DataFrame, y: Optional[pandas.Series] = None, coverage: List[float] = None, predictions: pandas.Series = None) Dict[str, pandas.Series] #
使用拟合的回归器查找预测区间。
此函数使用窗口大小为 5 的滚动标准差计算拟合估计器所有预测的滚动标准差。通过将各边界处下尾概率的百分点(分位数)函数乘以滚动标准差来确定预测的下限和上限。
- 参数
X (pd.DataFrame) – 数据,形状为 [n_samples, n_features]。
y (pd.Series) – 目标数据。忽略。
coverage (list[float]) – 0 到 1 之间的浮点数列表,用于计算预测区间的上限和下限。
predictions (pd.Series) – 可选的预测列表。如果为 None,则将使用 X 生成预测。
- 返回
预测区间,键格式为 {coverage}_lower 或 {coverage}_upper。
- 返回类型
dict
- 引发
MethodPropertyNotFoundError – 如果估计器不支持时间序列回归作为问题类型。
- static load(file_path)#
从文件路径加载组件。
- 参数
file_path (str) – 文件加载位置。
- 返回
ComponentBase object
- needs_fitting(self)#
返回布尔值,确定组件在调用 predict、predict_proba、transform 或 feature_importances 之前是否需要拟合。
对于不需要拟合或其拟合方法不执行任何操作的组件,可以将其重写为 False。
- 返回
True。
- property parameters(self)#
返回用于初始化组件的参数。
- predict(self, X)[source]#
使用拟合的 CatBoost 分类器进行预测。
- 参数
X (pd.DataFrame) – 数据,形状为 [n_samples, n_features]。
- 返回
预测值。
- 返回类型
pd.Series
- predict_proba(self, X)[source]#
使用拟合的 CatBoost 分类器进行概率预测。
- 参数
X (pd.DataFrame) – 数据,形状为 [n_samples, n_features]。
- 返回
预测概率值。
- 返回类型
pd.DataFrame
- save(self, file_path, pickle_protocol=cloudpickle.DEFAULT_PROTOCOL)#
将组件保存到文件路径。
- 参数
file_path (str) – 文件保存位置。
pickle_protocol (int) – pickle 数据流格式。
- update_parameters(self, update_dict, reset_fit=True)#
更新组件的参数字典。
- 参数
update_dict (dict) – 要更新的参数字典。
reset_fit (bool, optional) – 如果为 True,将把 _is_fitted 设置为 False。