Source code for libcity.data.dataset.abstract_dataset

[docs]class AbstractDataset(object): def __init__(self, config): raise NotImplementedError("Dataset not implemented")
[docs] def get_data(self): """ 返回数据的DataLoader,包括训练数据、测试数据、验证数据 Returns: tuple: tuple contains: train_dataloader: Dataloader composed of Batch (class) \n eval_dataloader: Dataloader composed of Batch (class) \n test_dataloader: Dataloader composed of Batch (class) """ raise NotImplementedError("get_data not implemented")
[docs] def get_data_feature(self): """ 返回一个 dict,包含数据集的相关特征 Returns: dict: 包含数据集的相关特征的字典 """ raise NotImplementedError("get_data_feature not implemented")