[發明專利]一種基于樹模型的預測方法和裝置

有效

• 申請號: 201911040223.X
• 申請日: 2019-10-29
• 公開（公告）號: CN110795603B
• 公開（公告）日: 2021-02-19
• 發明（設計）人: 陳超超;王力;周俊
• 申請（專利權）人: 支付寶（杭州）信息技術有限公司
• 主分類號: G06F16/901
• 分類號: G06F16/901;G06F21/62;G06Q10/04
• 代理公司: 北京億騰知識產權代理事務所(普通合伙) 11309
• 代理人: 張靜娟;周良玉

1.一種樹模型構建方法，其中，當前用于構建所述樹模型的構建數據由至少兩個數據方各自的設備擁有的數據組成，所述構建數據包括N個樣本各自的M個特征的特征值，所述至少兩個數據方中包括第一數據方和第二數據方，所述第一數據方的設備中具有N個樣本各自的標簽值，所述樹模型中當前包括第一樹，所述第一樹中當前包括第一節點，所述方法由所述第一數據方的設備相對于所述第一節點執行，包括：

從所述至少兩個數據方各自的設備獲取M組分裂結果，其中，所述M組分裂結果與所述M個特征分別對應，每組分裂結果中包括基于所述N個樣本各自的相應特征的特征值進行分裂的多個分裂結果；

記錄所述M組分裂結果各自對應的數據方；

基于N個樣本各自的標簽值，分別計算所述M組分裂結果中各個分裂結果的分裂增益；

獲取具有最大分裂增益的分裂結果作為最優分裂結果；

在所述最優分裂結果的分裂增益為正值的情況中，基于本地記錄的所述M組分裂結果各自對應的數據方，確定所述最優分裂結果對應的數據方；

在所述對應的數據方為所述第二數據方的情況中，將所述最優分裂結果發送給所述第二數據方的設備，并記錄所述第一節點與所述第二數據方的對應關系；

對所述第一節點進行標注，以指示本地沒有第一節點的節點數據，并相應地更新所述第一樹的樹結構。

2.根據權利要求1所述的方法，其中，所述M組分裂結果中的每組分裂結果包括與相應特征對應的P個分裂結果，其中P基于N個樣本中包括的該相應特征的非重復的特征值的個數確定。

3.根據權利要求1所述的方法，還包括，通知至少兩個數據方中除了所述第一數據方和所述第二數據方之外的其它數據方的設備，以指示所述其它數據方的設備對所述第一節點進行所述標注并相應地更新所述第一樹的樹結構。

4.根據權利要求1所述的方法，其中，所述第一數據方還具有N個樣本各自的第一特征的特征值，所述方法還包括，在所述對應的數據方是所述第一數據方的情況中，并且，在確定所述最優分裂結果為與第一特征對應的分裂結果的情況中，將所述第一特征確定為所述第一節點對應的特征，并將所述最優分裂結果對應的分裂值確定為所述第一節點的第一特征的分裂值，并通知所述至少兩個數據方中除所述第一數據方之外的其它數據方的設備，以指示其它數據方的設備對所述第一節點進行所述標注，并相應地更新所述第一樹的樹結構。

5.根據權利要求1所述的方法，還包括，在所述最優分裂結果的分裂增益小于等于零的情況中，基于所述N個樣本各自的標簽值計算所述第一節點對應的分值，并通知所述至少兩個數據方中除所述第一數據方之外的其它數據方的設備，以指示其它數據方的設備對所述第一節點進行所述標注。

6.根據權利要求1所述的方法，其中，所述第一樹為所述樹模型中的第t棵樹，其中，基于N個樣本各自的標簽值，分別計算所述M組分裂結果中各個分裂結果的分裂增益包括，基于N個樣本各自的標簽值以及預先獲取的所述樹模型的第t-1棵樹，分別計算所述M組分裂結果中各個分裂結果的分裂增益。

7.根據權利要求1所述的方法，其中，所述M組分裂結果中包括從至少兩個數據方各自的設備分別接收的重合的至少兩個分裂結果，基于本地記錄的所述M組分裂結果各自對應的數據方，確定所述最優分裂結果對應的數據方包括，在基于本地記錄的所述M組分裂結果各自對應的數據方，確定所述最優分裂結果為與至少兩個數據方對應的情況中，從所述至少兩個數據方中隨機確定一個數據方，作為所述最優分裂結果對應的數據方。

8.根據權利要求1所述的方法，其中，所述樹模型為以下任一種樹模型：Xgboost模型、GBDT模型、隨機森林。