模型訓(xùn)練方法、預(yù)測(cè)方法、裝置、設(shè)備及存儲(chǔ)介質(zhì)

基本信息

申請(qǐng)?zhí)?/td> CN202210310844.0 申請(qǐng)日 -
公開(kāi)(公告)號(hào) CN114638343A 公開(kāi)(公告)日 2022-06-17
申請(qǐng)公布號(hào) CN114638343A 申請(qǐng)公布日 2022-06-17
分類號(hào) G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06N20/20(2019.01)I 分類 計(jì)算;推算;計(jì)數(shù);
發(fā)明人 王若蘭;杜師帥;張鈞波;鄭宇 申請(qǐng)(專利權(quán))人 京東城市(北京)數(shù)字科技有限公司
代理機(jī)構(gòu) 北京同立鈞成知識(shí)產(chǎn)權(quán)代理有限公司 代理人 -
地址 100086北京市海淀區(qū)知春路76號(hào)(寫(xiě)字樓)1號(hào)樓9層1-7-5號(hào)
法律狀態(tài) -

摘要

摘要 本發(fā)明提供一種應(yīng)用于交易量預(yù)測(cè)的模型訓(xùn)練方法、預(yù)測(cè)方法、裝置、設(shè)備及存儲(chǔ)介質(zhì),該模型訓(xùn)練方法,包括:將樣本數(shù)據(jù)輸入各個(gè)應(yīng)用端的底層網(wǎng)絡(luò),得到狀態(tài)數(shù)據(jù),由服務(wù)端將各個(gè)底層網(wǎng)絡(luò)輸出的狀態(tài)數(shù)據(jù)拼接;將拼接狀態(tài)數(shù)據(jù)輸入第一應(yīng)用端的上層網(wǎng)絡(luò),得到預(yù)測(cè)交易量;根據(jù)預(yù)測(cè)交易量和實(shí)際交易量,生成第一梯度,并根據(jù)第一梯度更新第一應(yīng)用端的上層網(wǎng)絡(luò)的模型參數(shù);根據(jù)更新后的上層網(wǎng)絡(luò)以及第一梯度,生成第二梯度,并將第二梯度發(fā)送至服務(wù)端,服務(wù)端將第二梯度拆分為各個(gè)應(yīng)用端對(duì)應(yīng)的第三梯度,以基于第三梯度更新各個(gè)應(yīng)用端的底層網(wǎng)絡(luò)的模型參數(shù),直至交易量預(yù)測(cè)模型穩(wěn)定,基于多應(yīng)用端的數(shù)據(jù)進(jìn)行模型訓(xùn)練,提高了模型預(yù)測(cè)的準(zhǔn)確度。