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

基本信息

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

摘要

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