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