阿里推荐系统: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的错误,还有就是另外一种报错。因为这是开源的代码,所以代码应该是没有问题的,应该是数据集的问题。解决方法还没有找到。后续跑通的时候,会整理实验记录并且会更新后续博客。