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

基本信息

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

摘要

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