1. 首页
  2. 数据库
  3. 其它
  4. USTGCN:图神经网络的交通预测统一时空建模 源码

USTGCN:图神经网络的交通预测统一时空建模 源码

上传者: 2021-05-09 05:33:14上传 ZIP文件 23.52MB 热度 19次
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
下载地址
用户评论