libcity.data.dataset.dataset_subclass.multi_stgcnet_dataset

class libcity.data.dataset.dataset_subclass.multi_stgcnet_dataset.MultiSTGCnetDataset(config)[源代码]

基类:libcity.data.dataset.traffic_state_point_dataset.TrafficStatePointDataset, libcity.data.dataset.traffic_state_cpt_dataset.TrafficStateCPTDataset

get_data_feature()[源代码]

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

返回

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

返回类型

dict