对于某个节点,设其中所有样本的一阶梯度和为 G,二阶梯度和为 H,则节点分数为:
给定样本特征矩阵 X,以及每个样本的一阶梯度 g 和二阶梯度 h,请你实现 XGBoost 中基于 Split Gain 的最佳划分选择。
对于每个特征,需要枚举所有可能的划分阈值 threshold,将样本划分为左右两个子集:
X[i][feature] <= thresholdX[i][feature] > threshold设某个节点中所有样本的一阶梯度和为 G,二阶梯度和为 H,则该节点的分数为:
Scan the QR code below with WeChat to sign in
First-time scan will create your account automatically
请使用微信扫描下方二维码完成注册