libcity.data.dataset.dataset_subclass.hgcn_dataset

class libcity.data.dataset.dataset_subclass.hgcn_dataset.HGCNDataset(config)

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

calculate_adj_mx_cluster()

返回

#聚类结果[cluster_num][] 聚类标识[cluster_num][]

get_cluster()

返回

聚类后的中心向量矩阵，区域[节点]

get_data_feature()

返回数据集特征，scaler是归一化方法，adj_mx是邻接矩阵，num_nodes是点的个数， feature_dim是输入数据的维度，output_dim是模型输出的维度 :returns: 包含数据集的相关特征的字典 :rtype: dict