MSGIFSR代码复现过程中遇到的问题
文章:Learning Multi-granularity Consecutive User Intent Unit
for Session-based Recommendation
code地址:[(https://github.com/SpaceLearner/SessionRec-pytorch)
]
我是在华为云上跑的代码,配置镜像是:pytorch1.10-cuda10.2-cudnn7-ubuntu18.04
规格是:GPU: 1*V100(32GB)|CPU: 8核 64GB
按照GitHub中的提示运行命令:
配置环境:conda env create -f environment.yaml
代码中有三个模型的代码,所以分开运行
首先运行MSGIFER模型:
- 处理数据(以diginetica数据集为例)
python src/preprocess.py -d diginetica -f datasets/diginetica/train-item-views.csv - 运行代码:
python src/scripts/main_msgifsr.py(后面可以自己加上数据记得说明,也可以直接运行,这里运行的是sample数据集)
如果想要运行diginetica数据集的话,在src/srcipts/main_msgifer.py中将数据集改为diginetica即可。
运行过程中会出现ImportError: No module named 'src’的问题,解决办法如下:
代码中添加以下两行代码,然后删除src.的前缀即可:
import os.path
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))