№ CN121328646A発明
有効一种交互预测模型的训练方法、装置、设备及存储介质
本申请实施例提供了一种交互预测模型的训练方法、装置、设备及存储介质,可应用于图神经网络技术领域,该方法中针对以不同采样间隔采样的每个训练交互对序列,基于源节点在训练交互对序列中每个交互对的时间戳与验证预测时刻,确定第一时间特征矩阵;基于第一时间特征矩阵和训练交互对序列的采样间隔,确定在验证预测时刻源节点与任一交互节点的第一交互概率,从而根据在验证预测时刻源节点与任一交互节点的实际交互情况确定训练交互对序列的第一损失值;基于每个训练交互对序列的第一损失值,确定本轮迭代训练的综合损失值并对交互预测模型进行优化,引入时间特征提高准确性,而对不同采样间隔的训练提高了交互预测模型的泛化能力和鲁棒性。
❖上海上湖信息技术有限公司杨洋; 王扬; 祝溢泽; 王春平; 陈磊; 王玉翔§ 5/12 請求項G06N 3/0895 (2023.01), G06N 3/042 (2023.01), G06Q 50/20 (2012.01)12