libcity.data.dataset.traffic_state_od_dataset¶
-
class
libcity.data.dataset.traffic_state_od_dataset.
TrafficStateOdDataset
(config)[source]¶ Bases:
libcity.data.dataset.traffic_state_datatset.TrafficStateDataset
-
_add_external_information
(df, ext_data=None)[source]¶ 增加外部信息(一周中的星期几/day of week,一天中的某个时刻/time of day,外部数据), :param df: 交通状态数据多维数组, (len_time, …, feature_dim) :type df: np.ndarray :param ext_data: 外部数据 :type ext_data: np.ndarray
- Returns
融合后的外部数据和交通状态数据, (len_time, …, feature_dim_plus)
- Return type
np.ndarray
-