@@ -12,11 +12,13 @@ Here are two datasets we used in our paper. After downloaded the datasets, you c
1212
1313There 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
1719You 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
2224usage: preprocess.py [-h] [--dataset DATASET]
@@ -28,7 +30,7 @@ optional arguments:
2830
2931Then 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
3335You 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
7483Please cite our paper if you use the code:
0 commit comments