libcity.data.dataset.dataset_subclass.crann_dataset

class libcity.data.dataset.dataset_subclass.crann_dataset.CRANNDataset(config)[源代码]

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

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