可以在GPU上跑通的代码(含数据集),我已经放到了以下链接,
代码:
链接: https://pan.baidu.com/s/1_OKVH_UUwigVWOvdDVC_ig 提取码: 8a2z 复制这段内容后打开百度网盘手机App,操作更方便哦
数据集:
链接: https://pan.baidu.com/s/10ssrweMjrYBtNEIv4guE7A 提取码: 59qp 复制这段内容后打开百度网盘手机App,操作更方便哦
注意:我用的wadi数据集是删掉了几个归一化后列全为0的列,和原文不太一样;SWAT数据集是和原文一致的。
在服务器上跑
先创建一个虚拟环境
conda create -n GDN python==3.7
conda activate GDN
按照read me一步一步安装包。
### Requirements
* Python >= 3.6
* cuda == 10.2
* [Pytorch==1.5.1](https://pytorch.org/)要去这个网站复制下载命令
* [PyG: torch-geometric==1.5.0](https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html)
### Install packages
# run after installing correct Pytorch package
bash install.sh
安装
pip install scikit-learn==1.0.2
matplotlib == 3.1.1
numpy == 1.19.4
pandas == 0.25.1
bash run.sh cpu swat