libcity.data.dataset.dataset_subclass.gts_dataset

class libcity.data.dataset.dataset_subclass.gts_dataset.GTSDataset(config)[源代码]

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

get_data_feature()[源代码]

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

返回

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

返回类型

dict