《表3 Light GBM主要参数说明》
Light GBM主要采用直方图算法实现连续的浮点特征值离散化为k个整数离散值,并构造宽度为k的直方图,这样可以较大减少内存的消耗,减少运算的复杂度。此外,Ligh GBM使用带有深度限制的按叶子生长(leaf-wise)算法,即每次从当前所有叶子中找出增益最大的叶子节点进行分类,从而避免产生过多的误差,为防止产生过拟合增加了最大深度的限制。上述算法的关键参数可以在模型训练中指定和进行调试。LightGBM主要包含核心参数、控制参数、IO参数、目标参数、度量参数、网络参数等,在模型训练中常修改的便是核心参数、控制参数等(见表3)。
图表编号 | XD00196695600 严禁用于非法目的 |
---|---|
绘制时间 | 2020.11.15 |
作者 | 杨连报、刘新、董兴芝、李新琴、薛蕊、东春昭、徐贵红 |
绘制单位 | 中国铁道科学研究院集团有限公司电子计算技术研究所、中国铁道科学研究院集团有限公司运输及经济研究所、中国铁道科学研究院集团有限公司电子计算技术研究所、中国铁道科学研究院集团有限公司电子计算技术研究所、中国铁道科学研究院集团有限公司电子计算技术研究所、中国铁道科学研究院集团有限公司铁路基础设施检测中心、中国铁道科学研究院集团有限公司电子计算技术研究所 |
更多格式 | 高清、无水印(增值服务) |