交通预测数据文件梳理：PEMS-BAY

源远流长冬夏青青

已于 2024-08-05 22:13:46 修改

阅读量321

点赞数 5

文章标签：学习笔记数据挖掘

于 2024-08-05 22:09:51 首次发布

本文链接：https://blog.csdn.net/QAQ23333333333/article/details/140938132

版权

前言

最近做的实验比较多，对于交通预测数据的各种文件和文件中的数据格式理解愈加混乱，因此打算重新做一遍梳理来加深实验数据集的理解，本文章作为一个记录。

注意：本文主要以BasicTS中的数据文件作为例子，因为该工具库整理的数据和模型比较有体系，在此附上github链接和数据集下载路径。网盘密码：6v0a
BasicTS官方Github
BasicTS数据集百度网盘下载

该数据集主要包括以下文件（可以理解为图矩阵和特征矩阵两文件）
在这里插入图片描述

pkl文件包含三个子文件：

长度为325的graph_sensor_ids.txt文件存放了传感器的id，表示下标与传感器id一一对应
根据文件1的graph_sensor_ids.txt生成的 {sensor_id :index}的字典，表示下标与传感器id一一对应
一个325 * 325的带权非对称邻接矩阵dist_mx(有向图),其index通过上面的字典对应sensor_id,权值的计算原理是利用阈值高斯核，其中相当于做了归一化和稀疏化，将距离值映射到0-1之间，并将低于阈值的值置0。（权重是根据存放传感器之间的距离文件获取，该文件包含了{源传感器id，目标传感器id，cost代价}）

import pandas as pd
import numpy as np
df1 = pd.read_pickle("adj_PEMS-BAY.pkl")

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

52116条325个传感器的每五分钟检测交通速度值（特征矩阵）
交通速度取值范围：

这里读取文件的路径要修改为自己的。
在这里插入图片描述

关注