PinSage:首次在商业推荐系统成功应用的GCN模型实现分析

本文详细介绍了如何利用DGL框架实现PinSage模型,探讨了数据处理、图构建、节点特征生成、图卷积等关键步骤,并指出在实际运行中的一些注意事项和配置要求。

关注微信公众号【七步编程】,专注Python、AI、大数据,原创文章第一时间推送!

之前有过两篇文章分别介绍了GCN模型PinSage图神经网络框架DGL,本文就利用DGL来逐步实现PinSage模型,让熟悉DGL的使用过程中加深对PinSage的理解。

前言

上次的帖子分析了DGL这个框架,最后的地方提到DGL给出的示例中并不是所有的模型都是用消息传递的方式来编写的,比如著名的推荐模型PinSage,正巧这个模型在去年的帖子中也有过介绍(可见我这个人有多么拖沓,早就应该介绍模型实现,竟然拖了一年),PinSage并没有官方开源代码,所以这次就介绍一下DGL的实现方式,也给其他小伙伴一些启示吧。

如果对PinSage模型不太熟悉的,可以查看之前的一篇文章:PinSage:GCN在商业推荐系统首次成功应用

PinSage

首先模型在DGL项目的位置是:examples/pytorch/recommendation。模型是基于movielens数据集(https://grouplens.org/datasets/movielens/)实现的,数据中包含用户id,电影id,电影名字,时间戳,用户对电影评分等。DGL将数据集中用户和电影看作两种类型的节点,评分做为边,

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值