Skip to content
This repository was archived by the owner on Nov 12, 2021. It is now read-only.

Commit 90123c8

Browse files
committed
2 parents ad0bad9 + aa51065 commit 90123c8

1 file changed

Lines changed: 11 additions & 2 deletions

File tree

README.md

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,13 @@ Here are two datasets we used in our paper. After downloaded the datasets, you c
1212

1313
There is a small dataset `sample` included in the folder `datasets/`, which can be used to test the correctness of the code.
1414

15+
We have also written a [blog](https://sxkdz.github.io/research/SR-GNN) explaining the paper.
16+
1517
## Usage
1618

1719
You need to run the file `datasets/preprocess.py` first to preprocess the data.
1820

19-
For example: `cd datasets; python preprocess.py --dataset=sample`
21+
For example: `cd datasets; python preprocess.py --dataset=sample`
2022

2123
```bash
2224
usage: preprocess.py [-h] [--dataset DATASET]
@@ -28,7 +30,7 @@ optional arguments:
2830

2931
Then you can run the file `pytorch_code/main.py` or `tensorflow_code/main.py` to train the model.
3032

31-
For example: `cd pytorch_code; python main.py --dataset=sample`
33+
For example: `cd pytorch_code; python main.py --dataset=sample`
3234

3335
You can add the suffix `--nonhybrid` to use the global preference of a session graph to recommend instead of the hybrid preference.
3436

@@ -69,6 +71,13 @@ optional arguments:
6971
- Python 3
7072
- PyTorch 0.4.0 or Tensorflow 1.9.0
7173

74+
## Other Implementation for Reference
75+
There are other implementation available for reference:
76+
- Implementation based on PaddlePaddle by Baidu [[Link]](https://github.com/PaddlePaddle/models/tree/develop/PaddleRec/gnn)
77+
- Implementation based on PyTorch Geometric [[Link]](https://github.com/RuihongQiu/SR-GNN_PyTorch-Geometric)
78+
- Another implementation based on Tensorflow [[Link]](https://github.com/jimanvlad/SR-GNN)
79+
- Yet another implementation based on Tensorflow [[Link]](https://github.com/loserChen/TensorFlow-In-Practice/tree/master/SRGNN)
80+
7281
## Citation
7382

7483
Please cite our paper if you use the code:

0 commit comments

Comments
 (0)