一 代码解析

整体框架

 训练流程

1.初始化网络:class NetWork;

2.实例化实际的boosting树:依据参数选择不同模型,如GBDT、DART、GOSS、RF;

3.实例化实际的目标函数:依据参数选择不同类型,如RegressionL2loss、BinaryLogloss、MulticlassSoftmax、LambdarankNDCG等;

4.加载训练数据;

5.初始化实例化后boosting树;

6.初始化实例化后的目标函数;

7.训练模型并保存;

预测流程

细分模块

通信模块:

树结构

目标函数

数据加载

猜你喜欢

转载自www.cnblogs.com/bytedance/p/10346392.html
今日推荐