16 lines
830 B
Python
Executable File
16 lines
830 B
Python
Executable File
from dataloader.cde_loader.cdeDataloader import get_dataloader as cde_loader
|
|
from dataloader.PeMSDdataloader import get_dataloader as normal_loader
|
|
from dataloader.DCRNNdataloader import get_dataloader as DCRNN_loader
|
|
from dataloader.EXPdataloader import get_dataloader as EXP_loader
|
|
from dataloader.cde_loader.cdeDataloader import get_dataloader as nrde_loader
|
|
|
|
def get_dataloader(config, normalizer, single):
|
|
match config['model']['type']:
|
|
case 'STGNCDE': return cde_loader(config['data'], normalizer, single)
|
|
case 'STGNRDE': return nrde_loader(config['data'], normalizer, single)
|
|
case 'DCRNN': return DCRNN_loader(config['data'], normalizer, single)
|
|
case 'EXP': return EXP_loader(config['data'], normalizer, single)
|
|
case _: return normal_loader(config['data'], normalizer, single)
|
|
|
|
|