libcity.data.dataset.dataset_subclass.tgclstm_dataset¶

class libcity.data.dataset.dataset_subclass.tgclstm_dataset.TGCLSTMDataset(config)[source]¶

_load_rel()[source]¶: 加载.rel文件，格式[rel_id, type, origin_id, destination_id, properties(若干列)]

get_data_feature()[source]¶

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