文件名称:Neu-Review-Rec:使用Pytorch的基于神经审核的推荐模型的工具包
文件大小:30.71MB
文件格式:ZIP
更新时间:2024-05-22 13:17:43
pytorch recommendation-system daml review-based deepconn
带有Pytorch的基于神经审核的推荐模型的工具包。基于评论文本的深度推荐系统模型库(Pytorch) 更新:2021.03.29 添加一个分支(PL)以使用包装框架以进行进一步的分布式培训。 git clone https://github.com/ShomyLiu/Neu-Review-Rec.git git checkout pl 以及用法: python3 pl_main.py run --use_ddp=True --gpu_id=2 # indicating that using ddp mode for distributed training with 2 gpus. refer `config/config.py`. 基于神经评论的推荐 在这个仓库,我们重新实现一些重要的审核为基础的推荐模型,并提供与Pytorch可扩展的框架NRRec。 研究人员可以在我们的框架
【文件预览】:
Neu-Review-Rec-master
----.gitignore(190B)
----README.md(9KB)
----dataset()
--------data_review.py(1KB)
--------__init__.py(37B)
----models()
--------deepconn.py(2KB)
--------d_attn.py(4KB)
--------narre.py(4KB)
--------__init__.py(181B)
--------mpcn.py(6KB)
--------daml.py(5KB)
----main.py(7KB)
----checkpoints()
--------.gitkeeper(0B)
----pro_data()
--------Digital_Music_5.json.zip(30.69MB)
--------data_pro.py(17KB)
----config()
--------config.py(3KB)
--------__init__.py(892B)
----framework()
--------models.py(2KB)
--------fusion.py(2KB)
--------prediction.py(5KB)
--------__init__.py(50B)