libcity.data.dataset.dataset_subclass.staggcn_dataset

class libcity.data.dataset.dataset_subclass.staggcn_dataset.STAGGCNDataset(config)

基类：libcity.data.dataset.traffic_state_point_dataset.TrafficStatePointDataset

edge_index_func(matrix)

get_data_feature()

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

返回

包含数据集的相关特征的字典

返回类型

dict

get_dtw_edge_index()

get_edge_index()