【炼丹手记】在ModelArts上使用AI芯片Ascend训练基于MindSpore的图神经网络BGCF，并发布到AI Callery

论文

贝叶斯图协同过滤（BGCF）
论文原文如下
https://dl.acm.org/doi/pdf/10.1145/3394486.3403254

数据集

对Amazon-Beauty数据集进行预处理。
取得代码。通过git clone命令取得ModelZoo中已经实现好的代码

sh-4.4$git clone -b r1.5 --depth=1 https://gitee.com/mindspore/models.git
Cloning into 'models'...
remote: Enumerating objects: 7254, done.
remote: Counting objects: 100% (7254/7254), done.
remote: Compressing objects: 100% (4182/4182), done.
remote: Total 7254 (delta 3040), reused 5846 (delta 2670), pack-reused 0
Receiving objects: 100% (7254/7254), 59.11 MiB | 6.00 MiB/s, done.
Resolving deltas: 100% (3040/3040), done.
sh-4.4$

进入本模型的代码目录。可以发现预处理脚本在scripts下面。

sh-4.4$cd models/official/gnn/bgcf

于是，进入脚本所在目录

sh-4.4$cd scripts/

执行脚本。当然在执行之前，/cache/amazon下事先要放好原数据集文件。

sh-4.4$bash ./run_proces