SR-GNN

作者: 阿白coding的神 | 来源:发表于2022-11-21 22:57 被阅读0次

1.基本方法

1.将用户的行为序列构造 Session Graph

2.通过GNN来对所得的 Session Graph进行特征提取,得到每一个Item的向量表征

3.在经过GNN提取Session Graph之后,我们需要对所有的Item的向量表征进行融合,以此得到User的向量表征 在得到了用户的向量表征之后,我们就可以按照序列召回的思路来进行模型训练/模型验证了。

2. Graph的构建

这里利用有向图来构建,首先需要根据用户的行为序列。构图的方法也非常简单,如果v2和v1在用户的行为序列里面是相邻的,并且v2在v1之后,则我们连出一条从v2到v1的边。v1,v2都是item。

在完成构图之后,这里利用入度联合出度矩阵来构建。如下图所示

3. Graph信息传递

上式代表图中信息的传递,这是很标准的GNN信息传递,

实现如下:

4. 生成User 向量表征

代码实现如下

相关文章

  • SR-GNN

    1.基本方法 1.将用户的行为序列构造 Session Graph 2.通过GNN来对所得的 Session Gr...

网友评论

      本文标题:SR-GNN

      本文链接:https://www.haomeiwen.com/subject/edznxdtx.html