libcity.data.dataset.abstract_dataset¶

class libcity.data.dataset.abstract_dataset.AbstractDataset(config)[source]¶

Bases: object

get_data()[source]¶

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

Returns

tuple contains:

train_dataloader: Dataloader composed of Batch (class)

eval_dataloader: Dataloader composed of Batch (class)

test_dataloader: Dataloader composed of Batch (class)

Return type

tuple

get_data_feature()[source]¶

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