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()[源代码]