libcity.data.dataset.dataset_subclass.multi_stgcnet_dataset

class libcity.data.dataset.dataset_subclass.multi_stgcnet_dataset.MultiSTGCnetDataset(config)[source]

Bases: libcity.data.dataset.traffic_state_point_dataset.TrafficStatePointDataset, libcity.data.dataset.traffic_state_cpt_dataset.TrafficStateCPTDataset

get_data_feature()[source]

返回数据集特征,scaler是归一化方法,adj_mx是邻接矩阵,num_nodes是点的个数, feature_dim是输入数据的维度,output_dim是模型输出的维度, len_closeness/len_period/len_trend分别是三段数据的长度

Returns

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

Return type

dict