libcity.data.dataset.traffic_state_od_dataset¶

class libcity.data.dataset.traffic_state_od_dataset.TrafficStateOdDataset(config)[源代码]¶

基类：libcity.data.dataset.traffic_state_datatset.TrafficStateDataset

_add_external_information(df, ext_data=None)[源代码]¶

增加外部信息（一周中的星期几/day of week，一天中的某个时刻/time of day，外部数据）, :param df: 交通状态数据多维数组, (len_time, …, feature_dim) :type df: np.ndarray :param ext_data: 外部数据 :type ext_data: np.ndarray

返回: 融合后的外部数据和交通状态数据, (len_time, …, feature_dim_plus)
返回类型: np.ndarray

_load_geo()[源代码]¶: 加载.geo文件，格式[geo_id, type, coordinates, properties(若干列)]

_load_rel()[源代码]¶

加载.rel文件，格式[rel_id, type, origin_id, destination_id, properties(若干列)]

返回: self.adj_mx, N*N的邻接矩阵
返回类型: np.ndarray

get_data_feature()[源代码]¶

返回数据集特征，scaler是归一化方法，adj_mx是邻接矩阵，num_nodes是网格的个数， feature_dim是输入数据的维度，output_dim是模型输出的维度

返回: 包含数据集的相关特征的字典
返回类型: dict