阿里推荐系统:DSTN(一)

阿里推荐系统:dstn代码复现(一)

最近一直在复现阿里的一篇推荐系统文章,之前只是大概通读了一下。在复现代码的时候就选中了这篇文章,但是在复现的时候还是出现了一些问题,这也是我写这篇博客的原因,用来记录复现代码的过程。
代码地址
文章地址
以上给出了代码与文章的地址,感兴趣的可以跑一下。
在代码中给出了一个比较小的数据集,可以看成是样例。在readme中给出了数据集的地址,但是这个在kaggle上的数据集好像有一点问题,似乎需要处理一下,才能用具体的怎么处理我在后续的文章中会提到。
数据集地址
在代码中分别给出了DNN.py、dstn_int_att.py、dstn_pooling.py、dstn_self_att.py这四个不同方法的比较。
代码的环境要求是TensorFlow 1.3.0,因为目前我手里没有GPU,所以我只能用CPU跑代码。所幸在代码中附带的数据集是比较小的,用CPU跑起来还行。
但是我换成readme中提供的数据集的时候,就出现了outofrange的错误,还有就是另外一种报错。因为这是开源的代码,所以代码应该是没有问题的,应该是数据集的问题。解决方法还没有找到。后续跑通的时候,会整理实验记录并且会更新后续博客。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值