GraphRec_PyTorch GraphRec模型的PyTorch实现(Fan,Wenqi等人,“用于社交推荐的图神经网络。”万维网会议,ACM,2019年)。 用法 从requirements.txt文件安装所需的软件包。 pip install -r requirements.txt 预处理数据集。 应在数据集的相应文件夹中生成两个名为数据集和列表的pkl文件。 python preprocess.py --dataset Ciao python preprocess.py --dataset Epinions 运行main.py文件来训练模型。 您可以通过命令行配置一些训练参数。 python main.py 运行main.py文件以测试模型。 python main.py --test
1