文件名称:gcmc:重新实现“图卷积矩阵完成”(PyTorch和PyTorch Geometric)
文件大小:307KB
文件格式:ZIP
更新时间:2024-05-23 06:04:18
Python
图卷积矩阵完成(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.
【文件预览】:
gcmc-master
----.gitignore(1KB)
----src()
--------trainer.py(2KB)
--------layers.py(7KB)
--------utils.py(4KB)
--------dataset.py(4KB)
--------config.yml(365B)
--------model.py(5KB)
--------train.py(1KB)
----figs()
--------overview.png(165KB)
--------approach.png(158KB)
----README.md(1KB)
----requirements()
--------3.txt(23B)
--------1.txt(75B)
--------2.txt(87B)