├── LICENSE.txt ├── README.md ├── SAVE └── pretrain │ ├── GWN │ └── GWN_P8437.pth │ ├── MTGNN │ └── MTGNN_P8437.pth │ ├── PDFormer │ └── PDFormer_P8437.pth │ └── STGCN │ └── STGCN_P8437.pth ├── conf ├── AGCRN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf ├── ASTGCN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf ├── FlashST │ ├── Params_pretrain.py │ └── config.conf ├── GWN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS03.conf │ ├── PEMS04.conf │ ├── PEMS07.conf │ ├── PEMS07M.conf │ ├── PEMS08.conf │ └── chengdu_didi.conf ├── MSDR │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf ├── MTGNN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS03.conf │ ├── PEMS04.conf │ ├── PEMS07.conf │ ├── PEMS07M.conf │ ├── PEMS08.conf │ └── chengdu_didi.conf ├── PDFormer │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS03.conf │ ├── PEMS04.conf │ ├── PEMS07.conf │ ├── PEMS07M.conf │ ├── PEMS08.conf │ └── chengdu_didi.conf ├── ST-WA │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf ├── STFGNN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf ├── STGCN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS03.conf │ ├── PEMS04.conf │ ├── PEMS07.conf │ ├── PEMS07M.conf │ ├── PEMS08.conf │ └── chengdu_didi.conf ├── STSGCN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf └── TGCN │ ├── CA_District5.conf │ ├── NYC_BIKE.conf │ ├── PEMS07M.conf │ └── chengdu_didi.conf ├── lib ├── TrainInits.py ├── data_process.py ├── logger.py ├── metrics.py └── predifineGraph.py ├── model ├── AGCRN │ ├── AGCN.py │ ├── AGCRN.py │ ├── AGCRNCell.py │ └── args.py ├── ASTGCN │ ├── ASTGCN.py │ └── args.py ├── DMSTGCN │ └── DMSTGCN.py ├── FlashST.py ├── GWN │ ├── GWN.py │ └── args.py ├── MSDR │ ├── args.py │ ├── gmsdr_cell.py │ └── gmsdr_model.py ├── MTGNN │ ├── MTGNN.py │ └── args.py ├── PDFormer │ ├── PDFformer.py │ └── args.py ├── PromptNet.py ├── Run.py ├── STFGNN │ ├── STFGNN.py │ └── args.py ├── STGCN │ ├── args.py │ └── stgcn.py ├── STGODE │ ├── STGODE.py │ ├── args.py │ └── odegcn.py ├── STSGCN │ ├── STSGCN.py │ └── args.py ├── ST_WA │ ├── ST_WA.py │ ├── args.py │ └── attention.py ├── TGCN │ ├── TGCN.py │ └── args.py └── Trainer.py └── requirements.txt /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/README.md -------------------------------------------------------------------------------- /SAVE/pretrain/GWN/GWN_P8437.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/SAVE/pretrain/GWN/GWN_P8437.pth -------------------------------------------------------------------------------- /SAVE/pretrain/MTGNN/MTGNN_P8437.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/SAVE/pretrain/MTGNN/MTGNN_P8437.pth -------------------------------------------------------------------------------- /SAVE/pretrain/PDFormer/PDFormer_P8437.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/SAVE/pretrain/PDFormer/PDFormer_P8437.pth -------------------------------------------------------------------------------- /SAVE/pretrain/STGCN/STGCN_P8437.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/SAVE/pretrain/STGCN/STGCN_P8437.pth -------------------------------------------------------------------------------- /conf/AGCRN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/AGCRN/CA_District5.conf -------------------------------------------------------------------------------- /conf/AGCRN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/AGCRN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/AGCRN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/AGCRN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/AGCRN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/AGCRN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/ASTGCN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ASTGCN/CA_District5.conf -------------------------------------------------------------------------------- /conf/ASTGCN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ASTGCN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/ASTGCN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ASTGCN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/ASTGCN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ASTGCN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/FlashST/Params_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/FlashST/Params_pretrain.py -------------------------------------------------------------------------------- /conf/FlashST/config.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/FlashST/config.conf -------------------------------------------------------------------------------- /conf/GWN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/CA_District5.conf -------------------------------------------------------------------------------- /conf/GWN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/GWN/PEMS03.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/PEMS03.conf -------------------------------------------------------------------------------- /conf/GWN/PEMS04.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/PEMS04.conf -------------------------------------------------------------------------------- /conf/GWN/PEMS07.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/PEMS07.conf -------------------------------------------------------------------------------- /conf/GWN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/GWN/PEMS08.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/PEMS08.conf -------------------------------------------------------------------------------- /conf/GWN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/GWN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/MSDR/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MSDR/CA_District5.conf -------------------------------------------------------------------------------- /conf/MSDR/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MSDR/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/MSDR/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MSDR/PEMS07M.conf -------------------------------------------------------------------------------- /conf/MSDR/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MSDR/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/MTGNN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/CA_District5.conf -------------------------------------------------------------------------------- /conf/MTGNN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/MTGNN/PEMS03.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/PEMS03.conf -------------------------------------------------------------------------------- /conf/MTGNN/PEMS04.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/PEMS04.conf -------------------------------------------------------------------------------- /conf/MTGNN/PEMS07.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/PEMS07.conf -------------------------------------------------------------------------------- /conf/MTGNN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/MTGNN/PEMS08.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/PEMS08.conf -------------------------------------------------------------------------------- /conf/MTGNN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/MTGNN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/PDFormer/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/CA_District5.conf -------------------------------------------------------------------------------- /conf/PDFormer/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/PDFormer/PEMS03.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/PEMS03.conf -------------------------------------------------------------------------------- /conf/PDFormer/PEMS04.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/PEMS04.conf -------------------------------------------------------------------------------- /conf/PDFormer/PEMS07.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/PEMS07.conf -------------------------------------------------------------------------------- /conf/PDFormer/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/PEMS07M.conf -------------------------------------------------------------------------------- /conf/PDFormer/PEMS08.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/PEMS08.conf -------------------------------------------------------------------------------- /conf/PDFormer/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/PDFormer/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/ST-WA/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ST-WA/CA_District5.conf -------------------------------------------------------------------------------- /conf/ST-WA/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ST-WA/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/ST-WA/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ST-WA/PEMS07M.conf -------------------------------------------------------------------------------- /conf/ST-WA/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/ST-WA/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/STFGNN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STFGNN/CA_District5.conf -------------------------------------------------------------------------------- /conf/STFGNN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STFGNN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/STFGNN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STFGNN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/STFGNN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STFGNN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/STGCN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/CA_District5.conf -------------------------------------------------------------------------------- /conf/STGCN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/STGCN/PEMS03.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/PEMS03.conf -------------------------------------------------------------------------------- /conf/STGCN/PEMS04.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/PEMS04.conf -------------------------------------------------------------------------------- /conf/STGCN/PEMS07.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/PEMS07.conf -------------------------------------------------------------------------------- /conf/STGCN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/STGCN/PEMS08.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/PEMS08.conf -------------------------------------------------------------------------------- /conf/STGCN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STGCN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/STSGCN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STSGCN/CA_District5.conf -------------------------------------------------------------------------------- /conf/STSGCN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STSGCN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/STSGCN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STSGCN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/STSGCN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/STSGCN/chengdu_didi.conf -------------------------------------------------------------------------------- /conf/TGCN/CA_District5.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/TGCN/CA_District5.conf -------------------------------------------------------------------------------- /conf/TGCN/NYC_BIKE.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/TGCN/NYC_BIKE.conf -------------------------------------------------------------------------------- /conf/TGCN/PEMS07M.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/TGCN/PEMS07M.conf -------------------------------------------------------------------------------- /conf/TGCN/chengdu_didi.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/conf/TGCN/chengdu_didi.conf -------------------------------------------------------------------------------- /lib/TrainInits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/lib/TrainInits.py -------------------------------------------------------------------------------- /lib/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/lib/data_process.py -------------------------------------------------------------------------------- /lib/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/lib/logger.py -------------------------------------------------------------------------------- /lib/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/lib/metrics.py -------------------------------------------------------------------------------- /lib/predifineGraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/lib/predifineGraph.py -------------------------------------------------------------------------------- /model/AGCRN/AGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/AGCRN/AGCN.py -------------------------------------------------------------------------------- /model/AGCRN/AGCRN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/AGCRN/AGCRN.py -------------------------------------------------------------------------------- /model/AGCRN/AGCRNCell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/AGCRN/AGCRNCell.py -------------------------------------------------------------------------------- /model/AGCRN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/AGCRN/args.py -------------------------------------------------------------------------------- /model/ASTGCN/ASTGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/ASTGCN/ASTGCN.py -------------------------------------------------------------------------------- /model/ASTGCN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/ASTGCN/args.py -------------------------------------------------------------------------------- /model/DMSTGCN/DMSTGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/DMSTGCN/DMSTGCN.py -------------------------------------------------------------------------------- /model/FlashST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/FlashST.py -------------------------------------------------------------------------------- /model/GWN/GWN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/GWN/GWN.py -------------------------------------------------------------------------------- /model/GWN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/GWN/args.py -------------------------------------------------------------------------------- /model/MSDR/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/MSDR/args.py -------------------------------------------------------------------------------- /model/MSDR/gmsdr_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/MSDR/gmsdr_cell.py -------------------------------------------------------------------------------- /model/MSDR/gmsdr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/MSDR/gmsdr_model.py -------------------------------------------------------------------------------- /model/MTGNN/MTGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/MTGNN/MTGNN.py -------------------------------------------------------------------------------- /model/MTGNN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/MTGNN/args.py -------------------------------------------------------------------------------- /model/PDFormer/PDFformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/PDFormer/PDFformer.py -------------------------------------------------------------------------------- /model/PDFormer/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/PDFormer/args.py -------------------------------------------------------------------------------- /model/PromptNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/PromptNet.py -------------------------------------------------------------------------------- /model/Run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/Run.py -------------------------------------------------------------------------------- /model/STFGNN/STFGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STFGNN/STFGNN.py -------------------------------------------------------------------------------- /model/STFGNN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STFGNN/args.py -------------------------------------------------------------------------------- /model/STGCN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STGCN/args.py -------------------------------------------------------------------------------- /model/STGCN/stgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STGCN/stgcn.py -------------------------------------------------------------------------------- /model/STGODE/STGODE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STGODE/STGODE.py -------------------------------------------------------------------------------- /model/STGODE/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STGODE/args.py -------------------------------------------------------------------------------- /model/STGODE/odegcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STGODE/odegcn.py -------------------------------------------------------------------------------- /model/STSGCN/STSGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STSGCN/STSGCN.py -------------------------------------------------------------------------------- /model/STSGCN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/STSGCN/args.py -------------------------------------------------------------------------------- /model/ST_WA/ST_WA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/ST_WA/ST_WA.py -------------------------------------------------------------------------------- /model/ST_WA/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/ST_WA/args.py -------------------------------------------------------------------------------- /model/ST_WA/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/ST_WA/attention.py -------------------------------------------------------------------------------- /model/TGCN/TGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/TGCN/TGCN.py -------------------------------------------------------------------------------- /model/TGCN/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/TGCN/args.py -------------------------------------------------------------------------------- /model/Trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/model/Trainer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/FlashST/HEAD/requirements.txt --------------------------------------------------------------------------------