czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加STIDGCN

This commit is contained in:
czzhangheng 2025-04-23 23:24:43 +08:00
parent e826240a5e
commit 5f8c31af2e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model/model_selector.py
View File

@ -14,6 +14,7 @@ from model.STSGCN.STSGCN import STSGCN
from model.STGODE.STGODE import ODEGCN from model.STGODE.STGODE import ODEGCN
from model.PDG2SEQ.PDG2Seq import PDG2Seq from model.PDG2SEQ.PDG2Seq import PDG2Seq
from model.STMLP.STMLP import STMLP from model.STMLP.STMLP import STMLP
from model.STIDGCN.STIDGCN import STIDGCN
def model_selector(model): def model_selector(model):
match model['type']: match model['type']:
@ -33,4 +34,5 @@ def model_selector(model):
case 'STGODE': return ODEGCN(model) case 'STGODE': return ODEGCN(model)
case 'PDG2SEQ': return PDG2Seq(model) case 'PDG2SEQ': return PDG2Seq(model)
case 'STMLP': return STMLP(model) case 'STMLP': return STMLP(model)
case 'STIDGCN': return STIDGCN(model)