一种梯度提升树模型的预测方法、装置及系统与流程

文档序号:21004297发布日期:2020-06-05 22:58阅读:348来源:国知局
技术特征:

1.一种梯度提升树模型的预测方法,其特征在于,应用于第一数据方,所述方法包括:

基于待测用户的第一特征数据以及预先与第二数据方联合训练生成的梯度提升树模型,获取所述第一特征数据在所述梯度提升树模型中所对应的节点值;

接收所述第二数据方通过区块链发送的所述待测用户的第二特征数据在所述梯度提升树模型中所对应的节点值;

基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值以及所述梯度提升树模型,对所述待测用户进行标签预测。

2.根据权利要求1所述的方法,其特征在于,所述基于待测用户的第一特征数据以及预先与第二数据方联合训练生成的梯度提升树模型,获取所述第一特征数据在所述梯度提升树模型中所对应的节点值,包括:

提取所述梯度提升树模型中的回归树的所有特征切分变量以及各所述特征切分变量对应的切分点,并在所述所有特征切分变量中确定出属于所述第一数据方的目标特征切分变量;

将所述第一特征数据与所述目标特征切分变量对应的切分点进行比较,根据比较结果,获取所述第一特征数据在所述梯度提升树模型中对应的节点值。

3.根据权利要求1或2所述的方法,其特征在于,所述基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值以及所述梯度提升树模型,对所述待测用户进行标签预测,包括:

基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值,获取所述梯度提升树模型的每棵回归树对所述待测用户的预测值;

基于所述梯度提升树模型的每棵回归树对所述待测用户的预测值,获取所述待测用户的预测标签。

4.根据权利要求3所述的方法,其特征在于,所述基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值,获取所述梯度提升树模型的每棵回归树对所述待测用户的预测值,包括:

步骤a,将所述梯度提升树模型的基础回归树作为当前回归树;

步骤b,设置所述当前回归树的根节点为父节点;

步骤c,基于所述第一特征数据对应的节点值以及所述第二特征数据对应的节点值,在所述当前回归树中确定所述待测用户对应的子节点是所述父节点的左子节点还是所述父节点的右子节点,并执行步骤d;

步骤d,判断确定出的所述子节点是否为非叶子节点,若判断为是,则执行步骤e,否则,则执行步骤f;

步骤e,设置所述子节点为父节点,并跳转到步骤c;

步骤f,将所述子节点对应的预测值确定为所述当前回归树对所述待测用户的预测值,并执行步骤g;

步骤g,以所述当前回归树的下一棵回归树作为所述当前回归树,重复执行步骤b至步骤f,直至获得所述梯度提升树模型的最后一棵回归树对所述待测用户的预测值。

5.根据权利要求1所述的方法,其特征在于,所述方法还包括:

通过区块链发送所述待测用户的标签预测结果至所述第二数据方。

6.一种梯度提升树模型的预测方法,其特征在于,应用于第二数据方,所述方法包括:

在第一数据方获取到待测用户的第一特征数据在梯度提升树模型中所对应的节点值,并通过区块链将关于所述获取的通知信息发送给所述第二数据方之后,所述第二数据方基于所述待测用户的第二特征数据以及所述梯度提升树模型,获取所述第二特征数据在所述梯度提升树模型中所对应的节点值,其中,所述梯度提升树模型是所述第一数据方与所述第二数据方预先联合训练生成得到的;

将所述第二特征数据对应的节点值通过区块链发送给所述第一数据方,以使所述第一数据方基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值以及所述梯度提升树模型,对所述待测用户进行标签预测;

接收所述第一数据方通过区块链发送的所述待测用户的标签预测结果。

7.根据权利要求6所述的方法,其特征在于,所述第二数据方基于所述待测用户的第二特征数据以及所述梯度提升树模型,获取所述第二特征数据在所述梯度提升树模型中所对应的节点值,包括:

提取所述梯度提升树模型中的回归树的所有特征切分变量以及各所述特征切分变量对应的切分点,并在所述所有特征切分变量中确定出属于所述第二数据方的目标特征切分变量;

将所述第二特征数据与所述目标特征切分变量对应的切分点进行比较,根据比较结果,获取所述第二特征数据在所述梯度提升树模型中对应的节点值。

8.一种梯度提升树模型的预测装置,其特征在于,应用于第一数据方,所述装置包括:

获取模块,用于基于待测用户的第一特征数据以及所述第一数据方预先与第二数据方联合训练生成的梯度提升树模型,获取所述第一特征数据在所述梯度提升树模型中所对应的节点值;

接收模块,用于接收所述第二数据方通过区块链发送的所述待测用户的第二特征数据在所述梯度提升树模型中所对应的节点值;

预测模块,用于基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值以及所述梯度提升树模型,对所述待测用户进行标签预测。

9.一种梯度提升树模型的预测装置,其特征在于,应用于第二数据方,所述装置包括:

获取模块,用于在第一数据方获取到待测用户的第一特征数据在梯度提升树模型中所对应的节点值,并通过区块链将关于所述获取的通知信息发送给所述第二数据方之后,所述第二数据方基于所述待测用户的第二特征数据以及所述梯度提升树模型,获取所述第二特征数据在所述梯度提升树模型中所对应的节点值,其中,所述梯度提升树模型是所述第一数据方与所述第二数据方预先联合训练生成得到的;

发送模块,用于将所述第二特征数据对应的节点值通过区块链发送给所述第一数据方,以使所述第一数据方基于所述第一特征数据对应的节点值、所述第二特征数据对应的节点值以及所述梯度提升树模型,对所述待测用户进行标签预测;

接收模块,用于接收所述第一数据方通过区块链发送的所述待测用户的标签预测结果。

10.一种梯度提升树模型的预测系统,其特征在于,所述系统包括第一电子设备和第二电子设备,其中:

所述第一电子设备包括如权利要求8所述的梯度提升树模型的预测装置;

所述第二电子设备包括如权利要求9所述的梯度提升树模型的预测装置。


技术总结
本发明公开了一种梯度提升树模型的预测方法、装置及系统,属于机器学习领域,方法包括:第一数据方基于待测用户的第一特征数据以及预先与第二数据方联合训练生成的梯度提升树模型,获取第一特征数据在梯度提升树模型中所对应的节点值;接收第二数据方通过区块链发送的待测用户的第二特征数据在梯度提升树模型中所对应的节点值;基于第一特征数据对应的节点值、第二特征数据对应的节点值以及梯度提升树模型,对待测用户进行标签预测。本发明在实现梯度提升树模型的预测过程中,保证了两个数据方都不能获取对方的特征数据,避免了泄露双方持有的特征数据。

技术研发人员:姚雷;姚平;韩松江;李蒙;牛宝梅
受保护的技术使用者:苏宁云计算有限公司
技术研发日:2020.01.19
技术公布日:2020.06.05
当前第2页1 2 
网友询问留言 留言:0条
  • 还没有人留言评论。精彩留言会获得点赞!