pytorch == 1.3.1 numpy == 1.16.4 scipy == 1.4.1 scikit-learn == 0.22 运行代码 NGCF 戈瓦拉: python main.py --dataset gowalla --alg_type ngcf --regs [1e-5] --embed_size 64 --layer_size [64,64,64] --lr 0.0001 --save_flag 1 --pretrain 0 --batch_size 4096 --epoch 500 ...
这是我对本文的PyTorch实现: 王翔,何湘南,王萌,冯福利和蔡达生(2019)。 神经图协同过滤, 或。 在SIGIR'19,法国巴黎,2019年7月21日至25日。 TensorFlow实现可以在找到。 介绍 我的实现主要是指原始的TensorFlow实现。 它具有与原始项目一样的评估指标。 这是Gowalla数据集的示例: ...
liu-jc/PyTorch_NGCFPublic NotificationsYou must be signed in to change notification settings Fork7 Star27 master 3BranchesTags Code Folders and files Name Last commit message Last commit date Latest commit liu-jc init Mar 7, 2020 b45636e·Mar 7, 2020 ...
先hold一下https://github.com/huangtinglin/NGCF-PyTorch/blob/master/NGCF/utility/load_data.py算法小屋 文章被收录于专栏 不定期分享各类算法以及面经。同时也正在学习相关分布式技术。欢迎一起交流。全部评论 推荐 最新 楼层在思考的嘟教授很无聊 厦门大学 产品经理 你好,关于NGCF有些问题想请教一下可以吗?qq...
需要着重说明的是\color{red}{需要着重说明的是}需要着重说明的是以下log是对齐pytorch时的训练结果,但后期我们查看论文tensorflow版本代码时,发现未与tensorflow原版对齐,因此在介绍的notebook中,会说明最后我们提交精度是与tensorflow对齐的,本项目代码是与tensorflow对齐的最新版本,这也是在开头强调要加载最新版本的原因。
Data init Mar 7, 2020 NGCF init Mar 7, 2020 .gitignore init Mar 7, 2020 README.md init Mar 7, 2020 requirements.txt init Mar 7, 2020 This is a PyTorch Implemenation for this paper: Xiang Wang, Xiangnan He, Meng Wang, Fuli Feng, and Tat-Seng Chua (2019). Neural Graph Collaborati...