一種GBDT模型的縱向訓(xùn)練方法、裝置、電子設(shè)備及系統(tǒng)

基本信息

申請?zhí)?/td> CN202210110534.4 申請日 -
公開（公告）號 CN114118312A 公開（公告）日 2022-03-01
申請公布號 CN114118312A 申請公布日 2022-03-01
分類號 G06K9/62（2022.01）I 分類 計算；推算；計數(shù);
發(fā)明人 郝天一;陳智隆;陳琨;王國賽 申請（專利權(quán)）人 華控清交信息科技（北京）有限公司
代理機(jī)構(gòu) 北京市廣友專利事務(wù)所有限責(zé)任公司 代理人 李欣
地址 100084北京市海淀區(qū)中關(guān)村東路1號院3號樓10層1009-1
法律狀態(tài) -

摘要

摘要 本申請公開了一種GBDT模型的縱向訓(xùn)練方法、裝置、電子設(shè)備及系統(tǒng)，包括：針對初始GBDT模型的當(dāng)前決策樹的當(dāng)前層，獲取多個樣本各自位于當(dāng)前層的節(jié)點的節(jié)點編號；分別針對當(dāng)前層的每個節(jié)點，基于多個樣本各自的節(jié)點編號，按照自身持有的每種特征的分箱數(shù)，對該節(jié)點包括的所有樣本進(jìn)行分箱，得到分箱結(jié)果；針對能夠用于進(jìn)行當(dāng)前層的節(jié)點分裂的各種分裂標(biāo)準(zhǔn)，分別計算按照每種分裂標(biāo)準(zhǔn)進(jìn)行節(jié)點分裂的評分增益值，評分增益值最高的分裂標(biāo)準(zhǔn)，作為當(dāng)前層的分裂標(biāo)準(zhǔn)；按照該分裂標(biāo)準(zhǔn)，進(jìn)行節(jié)點分裂，得到當(dāng)前層的節(jié)點分裂結(jié)果，完成GBDT模型訓(xùn)練。采用該方案，在各數(shù)據(jù)節(jié)點不泄露自身的特征數(shù)據(jù)的情況下，實現(xiàn)了針對GBDT模型的縱向訓(xùn)練。