libcity.data.dataset.dataset_subclass.chebconv_dataset

class libcity.data.dataset.dataset_subclass.chebconv_dataset.ChebConvDataset(config)

Bases: libcity.data.dataset.abstract_dataset.AbstractDataset

_get_scalar(scaler_type, data)

根据全局参数`scaler_type`选择数据归一化方法

Parameters

data – 训练数据X

Returns

归一化对象

Return type

Scaler

_load_cache_train_val_test()

加载之前缓存好的训练集、测试集、验证集

_load_geo()

加载.geo文件，格式[geo_id, type, coordinates, properties(若干列)]

_load_rel()

加载.rel文件，格式[rel_id, type, origin_id, destination_id, properties(若干列)], 生成N*N的矩阵，默认.rel存在的边表示为1，不存在的边表示为0

Returns

self.adj_mx, N*N的邻接矩阵

Return type

np.ndarray

get_data()

返回数据的DataLoader，包括训练数据、测试数据、验证数据

Returns

dict

Return type

batch_data

get_data_feature()

返回一个 dict，包含数据集的相关特征

Returns

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

Return type

dict