XGB-10: 多输出multioutput

自版本1.6开始,XGBoost的Python包具有对多输出回归和多标签分类的实验性支持多标签分类通常指具有多个非排他性类标签的目标。例如,电影可以同时被分类为科幻和喜剧。

使用单模型处理多目标训练

默认情况下,XGBoost为每个目标构建一个模型,类似于sklearn的元估计器(meta estimators),同时具有重用数据和其他集成功能(如SHAP)的优势。有关回归的示例,请参阅多输出回归演示对于多标签分类,使用二进制相关策略。输入y的形状应为(n_samples,n_classes),每列的值为0或1,以指定样本是否标记为相应的类别。给定一个具有3个输出类和2个标签的样本,相应的y应编码为[1, 0, 1],其中第二类标签为负,其余为正。目前,XGBoost仅支持标签的稠密矩阵。

from sklearn.datasets import make_multilabel_classification
import numpy as np

X, y = make_multilabel_classification(
    n_samples=32, n_classes=5, n_labels=3, random_state=0
)

clf = xgb.XGBClassifier(tree_method="hist")
clf.fit(X, y)
np.testing.assert_allclose(clf.predict(X), y)

使用 Vector Leaf 进行训练

XGBoost可以选择使用hist tree方法构建多输出树,其中叶子的大小等于目标的数量。这种行为可以通过multi_strategy训练参数进行控制,该参数可以取值one_output_per_tree(默认值),用于为每个目标构建一个模型,或者multi_output_tree,用于构建多输出树。

clf = xgb.XGBClassifier(tree_method="hist", multi_strategy="multi_output_tree")

参考

  • 7
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

uncle_ll

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值