一种GBDT模型的纵向训练方法、装置、电子设备及系统

基本信息

申请号 CN202210110534.4 申请日 -
公开(公告)号 CN114118312A 公开(公告)日 2022-03-01
申请公布号 CN114118312A 申请公布日 2022-03-01
分类号 G06K9/62(2022.01)I 分类 计算;推算;计数;
发明人 郝天一;陈智隆;陈琨;王国赛 申请(专利权)人 华控清交信息科技(北京)有限公司
代理机构 北京市广友专利事务所有限责任公司 代理人 李欣
地址 100084北京市海淀区中关村东路1号院3号楼10层1009-1
法律状态 -

摘要

摘要 本申请公开了一种GBDT模型的纵向训练方法、装置、电子设备及系统,包括:针对初始GBDT模型的当前决策树的当前层,获取多个样本各自位于当前层的节点的节点编号;分别针对当前层的每个节点,基于多个样本各自的节点编号,按照自身持有的每种特征的分箱数,对该节点包括的所有样本进行分箱,得到分箱结果;针对能够用于进行当前层的节点分裂的各种分裂标准,分别计算按照每种分裂标准进行节点分裂的评分增益值,评分增益值最高的分裂标准,作为当前层的分裂标准;按照该分裂标准,进行节点分裂,得到当前层的节点分裂结果,完成GBDT模型训练。采用该方案,在各数据节点不泄露自身的特征数据的情况下,实现了针对GBDT模型的纵向训练。