libcity.data.dataset.dataset_subclass.gts_dataset

class libcity.data.dataset.dataset_subclass.gts_dataset.GTSDataset(config)[source]

Bases: libcity.data.dataset.traffic_state_point_dataset.TrafficStatePointDataset

get_data_feature()[source]

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

Returns

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

Return type

dict