57 lines
2.3 KiB
Python
Executable File
57 lines
2.3 KiB
Python
Executable File
from model.DDGCRN.DDGCRN import DDGCRN
|
|
from model.TWDGCN.TWDGCN import TWDGCN
|
|
from model.AGCRN.AGCRN import AGCRN
|
|
from model.NLT.HierAttnLstm import HierAttnLstm
|
|
from model.STGNCDE.Make_model import make_model
|
|
from model.DSANET.DSANET import DSANet
|
|
from model.STGCN.models import STGCNChebGraphConv
|
|
from model.DCRNN.dcrnn_model import DCRNNModel
|
|
from model.ARIMA.ARIMA import ARIMA
|
|
from model.TCN.TCN import TemporalConvNet
|
|
from model.GWN.GraphWaveNet import gwnet
|
|
from model.STFGNN.STFGNN import STFGNN
|
|
from model.STSGCN.STSGCN import STSGCN
|
|
from model.STGODE.STGODE import ODEGCN
|
|
from model.PDG2SEQ.PDG2Seqb import PDG2Seq
|
|
from model.STMLP.STMLP import STMLP
|
|
from model.STIDGCN.STIDGCN import STIDGCN
|
|
from model.STID.STID import STID
|
|
from model.STAEFormer.STAEFormer import STAEformer
|
|
from model.EXP.EXP32 import EXP as EXP
|
|
from model.MegaCRN.MegaCRNModel import MegaCRNModel
|
|
from model.ST_SSL.ST_SSL import STSSLModel
|
|
from model.STGNRDE.Make_model import make_model as make_nrde_model
|
|
from model.STAWnet.STAWnet import STAWnet
|
|
from model.STEP.STEP import STEP
|
|
from model.RGDAN.RGDAN import RGDANModel
|
|
|
|
def model_selector(model):
|
|
match model['type']:
|
|
case 'DDGCRN': return DDGCRN(model)
|
|
case 'TWDGCN': return TWDGCN(model)
|
|
case 'AGCRN': return AGCRN(model)
|
|
case 'NLT': return HierAttnLstm(model)
|
|
case 'STGNCDE': return make_model(model)
|
|
case 'DSANET': return DSANet(model)
|
|
case 'STGCN': return STGCNChebGraphConv(model)
|
|
case 'DCRNN': return DCRNNModel(model)
|
|
case 'ARIMA': return ARIMA(model)
|
|
case 'TCN': return TemporalConvNet(model)
|
|
case 'GWN': return gwnet(model)
|
|
case 'STFGNN': return STFGNN(model)
|
|
case 'STSGCN': return STSGCN(model)
|
|
case 'STGODE': return ODEGCN(model)
|
|
case 'PDG2SEQ': return PDG2Seq(model)
|
|
case 'STMLP': return STMLP(model)
|
|
case 'STIDGCN': return STIDGCN(model)
|
|
case 'STID': return STID(model)
|
|
case 'STAEFormer': return STAEformer(model)
|
|
case 'EXP': return EXP(model)
|
|
case 'MegaCRN': return MegaCRNModel(model)
|
|
case 'ST_SSL': return STSSLModel(model)
|
|
case 'STGNRDE': return make_nrde_model(model)
|
|
case 'STAWnet': return STAWnet(model)
|
|
case 'STEP': return STEP(model)
|
|
case 'RGDAN': return RGDANModel(model)
|
|
|