USTGCN:图神经网络的交通预测统一时空建模 源码
USTGCN:使用图形神经网络进行交通预测的统一时空建模 作者 这是IJCNN 2021接受的“使用图形神经网络进行交通预测的统一时空建模”的pytorch实现。 环境设置 克隆git项目: $ git clone https://github.com/AmitRoy7781/USTGCN 创建一个新的conda环境并安装所需的软件包(命令适用于ubuntu 16.04) $ conda create -n TrafficEnv python=3.7 $ conda activate TrafficEnv $ pip install -r requirements.txt 基本用法: 主要参数: --dataset The input traffic dataset(default:PeMSD7) --GNN_layers Number of l
下载地址
用户评论