模型训练方法、预测方法、装置、设备及存储介质
基本信息
申请号 | 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 | 分类 | 计算;推算;计数; |
发明人 | 王若兰;杜师帅;张钧波;郑宇 | 申请(专利权)人 | 京东城市(北京)数字科技有限公司 |
代理机构 | 北京同立钧成知识产权代理有限公司 | 代理人 | - |
地址 | 100086北京市海淀区知春路76号(写字楼)1号楼9层1-7-5号 | ||
法律状态 | - |
摘要
摘要 | 本发明提供一种应用于交易量预测的模型训练方法、预测方法、装置、设备及存储介质,该模型训练方法,包括:将样本数据输入各个应用端的底层网络,得到状态数据,由服务端将各个底层网络输出的状态数据拼接;将拼接状态数据输入第一应用端的上层网络,得到预测交易量;根据预测交易量和实际交易量,生成第一梯度,并根据第一梯度更新第一应用端的上层网络的模型参数;根据更新后的上层网络以及第一梯度,生成第二梯度,并将第二梯度发送至服务端,服务端将第二梯度拆分为各个应用端对应的第三梯度,以基于第三梯度更新各个应用端的底层网络的模型参数,直至交易量预测模型稳定,基于多应用端的数据进行模型训练,提高了模型预测的准确度。 |
