libcity.data.dataset.abstract_dataset

class libcity.data.dataset.abstract_dataset.AbstractDataset(config)[源代码]

基类:object

get_data()[源代码]

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

返回

tuple contains:

train_dataloader: Dataloader composed of Batch (class)

eval_dataloader: Dataloader composed of Batch (class)

test_dataloader: Dataloader composed of Batch (class)

返回类型

tuple

get_data_feature()[源代码]

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

返回

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

返回类型

dict