文件名称:RecSystem-papers-Implemention-master:PyTorch和tf2.x实现深度学习RecSystem模型的实现
文件大小:33KB
文件格式:ZIP
更新时间:2024-05-28 04:23:26
Python
Implements of Deep Learning RecSystem Model with PyTorch and tf2.x 最近从CV转行做推荐,阅读了许多用深度学习做推荐的paper,有感而发,觉得推荐领域论文的工程性大都很强,很多都是从实际的业务和数据出发,在解决业务需求的同时能产出一篇不错的paper。同时在自己学习的过程中觉得paper上读到的东西终究有限,需要自己实现一下论文方法,才能有更加深刻的理解。 Introduction 本项目旨在对自己读过的和感兴趣的深度学习推荐系统领域的文章进行复现 提供PyTorch和Tensorflow2.x两种框架的实现 使用lmdb数据库加速数据加载过程 支持单GPU和多GPU训练加速 Dataset Ad Display/Click data on Taobao.com[TODO] Some importent detai
【文件预览】:
RecSystem-papers-Implemention-master-main
----train_example_pytorch.py(7KB)
----model()
--------DeepFM()
--------AFM()
--------FFM()
--------xDeepFM()
--------DeepCross()
--------Autoint()
--------FM()
----dataset()
--------Criteo.py(4KB)
--------Criteo.md(2KB)
----train_example_tensorflow.py(5KB)
----README.md(6KB)
----Layers()
--------Layers_PyTorch.py(7KB)
--------Layers_Tensorflow.py(10KB)