STGCN_IJCAI-18在服务器tensorflow 2.5.0下运行

源码github地址：https://github.com/VeritasYin/STGCN_IJCAI-18

服务器环境：
tensorflow 2.5.0
python 3.8.3
numpy 1.19.5

1.首先在STGCN_IJCAI-18下新建一个文件夹dataset，并将data_loader中的PeMS-M解压缩到该文件夹中；

2.将main.py中的import tensorflow as tf改成

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

3.对main,py中改为下图这样（这个文件取决于你dataset中的文件的名字，不想改代码的话把dataset中的两个文件改成代码中的名字也可以）

4.将models/trainer.py，models/layers.py，models/base_model.py中的import tensorflow as tf改成

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

5.运行

python main.py --n_route 228 --graph W_228.csv