libcity.data.dataset.traffic_state_point_dataset¶
-
class
libcity.data.dataset.traffic_state_point_dataset.TrafficStatePointDataset(config)[source]¶ Bases:
libcity.data.dataset.traffic_state_datatset.TrafficStateDataset-
_add_external_information(df, ext_data=None)[source]¶ 增加外部信息（一周中的星期几/day of week，一天中的某个时刻/time of day，外部数据）
- Parameters
df (np.ndarray) – 交通状态数据多维数组, (len_time, num_nodes, feature_dim)
ext_data (np.ndarray) – 外部数据
- Returns
融合后的外部数据和交通状态数据, (len_time, num_nodes, feature_dim_plus)
- Return type
np.ndarray
-
_load_dyna(filename)[source]¶ 加载.dyna文件，格式[dyna_id, type, time, entity_id, properties(若干列)] 其中全局参数`data_col`用于指定需要加载的数据的列，不设置则默认全部加载
- Parameters
filename (str) – 数据文件名，不包含后缀
- Returns
数据数组, 3d-array (len_time, num_nodes, feature_dim)
- Return type
np.ndarray
-