图卷积矩阵完成(Pytorch)
重新实现(PyTorch和PyTorch Geometric)
笔记
该存储库不是本文的正式实现。 官方的实现是(Tensorflow)。 我们的实验结果如下所示,但未达到原始分数。
设置
使用python 3.6或更高版本设置虚拟环境
安装要求(点子) pip install -r requirements/1.txt
pip install --verbose --no-cache-dir -r requirements/2.txt
pip install -r requirements/3.txt
安装Pytorch Geometric非常麻烦,可能会破坏您的python环境。 因此,我们强烈建议您使用虚拟环境(例如pyenv,virtualenv,pipenv等)。 有关更多详细信息,请参见。
训练与测试
cd src
python train.