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.PDG2SEQ.PDG2Seq import PDG2Seq
from model.STMLP.STMLP import STMLP
from model.STIDGCN.STIDGCN import STIDGCN
def model_selector(model):
match model['type']:
@ -33,4 +34,5 @@ def model_selector(model):
case 'STGODE': return ODEGCN(model)
case 'PDG2SEQ': return PDG2Seq(model)
case 'STMLP': return STMLP(model)
case 'STIDGCN': return STIDGCN(model)