未验证 提交 72ee1ad6 编写于 作者: H Huang Zhengjie 提交者: GitHub

Merge pull request #50 from Liwb5/develop

udpate README in GIN
...@@ -4,7 +4,8 @@ ...@@ -4,7 +4,8 @@
### Datasets ### Datasets
The dataset can be downloaded from [here](https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip) The dataset can be downloaded from [here](https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip).
After downloading the data,uncompress them, then a directory named `./dataset/` can be found in current directory. Note that the current directory is the root directory of GIN model.
### Dependencies ### Dependencies
...@@ -15,7 +16,7 @@ The dataset can be downloaded from [here](https://github.com/weihua916/powerful- ...@@ -15,7 +16,7 @@ The dataset can be downloaded from [here](https://github.com/weihua916/powerful-
For examples, use GPU to train GIN model on MUTAG dataset. For examples, use GPU to train GIN model on MUTAG dataset.
``` ```
python main.py --use_cuda --dataset_name MUTAG python main.py --use_cuda --dataset_name MUTAG --data_path ./dataset
``` ```
### Hyperparameters ### Hyperparameters
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册