検索に戻る
案件記録

一种交互预测模型的训练方法、装置、设备及存储介质

発明有効
1閲覧数
12請求項 · 5 独立
§ Ⅰ

案件概要

発明者

杨洋; 王扬; 祝溢泽; 王春平; 陈磊; 王玉翔

IPC分類

G06N 3/0895 (2023.01)G06N 3/042 (2023.01)G06Q 50/20 (2012.01)

本申请实施例提供了一种交互预测模型的训练方法、装置、设备及存储介质,可应用于图神经网络技术领域,该方法中针对以不同采样间隔采样的每个训练交互对序列,基于源节点在训练交互对序列中每个交互对的时间戳与验证预测时刻,确定第一时间特征矩阵;基于第一时间特征矩阵和训练交互对序列的采样间隔,确定在验证预测时刻源节点与任一交互节点的第一交互概率,从而根据在验证预测时刻源节点与任一交互节点的实际交互情况确定训练交互对序列的第一损失值;基于每个训练交互对序列的第一损失值,确定本轮迭代训练的综合损失值并对交互预测模型进行优化,引入时间特征提高准确性,而对不同采样间隔的训练提高了交互预测模型的泛化能力和鲁棒性。

外部リソース