
GCN_Predict-Pytorch: 基于PyTorch的交通流量预测模型图卷积网络实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
GCN_Predict-Pytorch是一个基于PyTorch框架开发的开源项目,用于实现交通流量预测的图卷积神经网络模型。该工具为研究人员和开发者提供了一个高效、灵活的平台来探索和应用先进的机器学习技术解决交通领域的问题。
使用PyTorch实现图卷积网络(GCN、GAT、Chebnet)进行交通流量预测。
数据集由Caltrans绩效评估系统(PEMS-04)收集,包含307个探测器的数据,时间范围为2018年1月至2月。特征包括流动量、占有率和速度。
探索性数据分析:
1. 数据具有三个特点:流动量、占用率和速度。首先对数据分布进行可视化分析。
2. 运行代码以查看数据:`python data_view.py`
3. 每个节点(探测器)有三个特征,但两个特征的数据分布相对固定,因此我们只采用一维特征。
读取数据集:
在traffic_dataset.py文件中,使用get_adjacent_matrix和get_flow_data函数来获取相邻矩阵和流量数据。
模型训练:相关的代码位于tra目录下。
全部评论 (0)
还没有任何评论哟~


