libcity.data.dataset.traffic_state_grid_od_dataset¶
-
class
libcity.data.dataset.traffic_state_grid_od_dataset.TrafficStateGridOdDataset(config)[source]¶ Bases:
libcity.data.dataset.traffic_state_datatset.TrafficStateDataset-
_add_external_information(df, ext_data=None)[source]¶ 增加外部信息（一周中的星期几/day of week，一天中的某个时刻/time of day，外部数据）, 根据参数`use_row_column`确定是4d还是6d的数组，True为6d
- Parameters
df (np.ndarray) – 交通状态数据多维数组, (len_time, …, feature_dim)
ext_data (np.ndarray) – 外部数据
- Returns
融合后的外部数据和交通状态数据, (len_time, …, feature_dim_plus)
- Return type
np.ndarray
-
_load_dyna(filename)[source]¶ 加载.gridod文件，格式[dyna_id, type, time, origin_row_id, origin_column_id, destination_row_id, destination_column_id, properties(若干列)], 其中全局参数`data_col`用于指定需要加载的数据的列，不设置则默认全部加载, 根据参数`use_row_column`确定转成4d还是6d的数组，True为6d
- Parameters
filename (str) – 数据文件名，不包含后缀
- Returns
数据数组, 4d-array or 6d-array (len_time, num_grids, num_grids, feature_dim) / (len_time, len_row, len_column, len_row, len_column, feature_dim)
- Return type
np.ndarray
-