上传者: 42099176
|
上传时间: 2022-01-18 15:10:52
|
文件大小: 23.52MB
|
文件类型: -
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