libcity.data.dataset.crann_dataset

class libcity.data.dataset.crann_dataset.CRANNDataset(config)[source]

Bases: libcity.data.dataset.traffic_state_point_dataset.TrafficStatePointDataset

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