├── LICENSE ├── Params.py ├── README.md ├── componenets ├── __pycache__ │ ├── attention.cpython-38.pyc │ ├── engine.cpython-38.pyc │ ├── metrics.cpython-310.pyc │ ├── metrics.cpython-38.pyc │ ├── metrics.cpython-39.pyc │ ├── model.cpython-38.pyc │ ├── new_metrics.cpython-310.pyc │ ├── new_metrics.cpython-38.pyc │ ├── new_metrics.cpython-39.pyc │ └── normalization.cpython-38.pyc ├── metrics.py ├── new_metrics.py └── normalization.py ├── config └── STExplainer_pems4.yaml ├── data ├── DataHandler_st.py ├── PEMS04 │ ├── PEMS04.csv │ └── PEMS04.npz └── __pycache__ │ ├── DataHandler.cpython-38.pyc │ ├── DataHandler_pyg.cpython-38.pyc │ ├── DataHandler_st.cpython-38.pyc │ ├── STGDataset.cpython-38.pyc │ └── STGDataset_pyg.cpython-38.pyc ├── engine.py ├── images ├── .DS_Store └── STExplainer.png ├── methods └── STExplainer │ ├── MLP.py │ ├── __pycache__ │ ├── MLP.cpython-310.pyc │ ├── MLP.cpython-38.pyc │ ├── MLP.cpython-39.pyc │ ├── STID_arch.cpython-38.pyc │ ├── STID_mo.cpython-38.pyc │ ├── gat_gib.cpython-310.pyc │ ├── gat_gib.cpython-38.pyc │ ├── gatconv_gib.cpython-310.pyc │ ├── gatconv_gib.cpython-38.pyc │ ├── model_utils.cpython-38.pyc │ ├── stgib.cpython-38.pyc │ ├── stgib_series.cpython-310.pyc │ ├── stgib_series.cpython-38.pyc │ └── stgib_series.cpython-39.pyc │ ├── gat_gib.py │ ├── gatconv_gib.py │ ├── model_utils.py │ ├── stgat │ ├── MLP.py │ ├── __pycache__ │ │ ├── MLP.cpython-38.pyc │ │ ├── conv_layers.cpython-38.pyc │ │ ├── gat.cpython-38.pyc │ │ └── stgat.cpython-38.pyc │ ├── conv_layers.py │ ├── gat.py │ └── stgat.py │ ├── stgib.py │ ├── stgib_series.py │ └── stgsat │ ├── MLP.py │ ├── __pycache__ │ ├── MLP.cpython-38.pyc │ ├── conv_layers.cpython-38.pyc │ ├── gat_ib.cpython-38.pyc │ ├── get_model.cpython-38.pyc │ ├── gsat.cpython-38.pyc │ └── stgsat.cpython-38.pyc │ ├── conv_layers.py │ ├── gat_ib.py │ ├── get_model.py │ ├── gsat.py │ └── stgsat.py ├── requirements.txt ├── results └── model │ └── 2023y10m24d │ └── STExplainer_pems4.pkl ├── train.py └── utils ├── __pycache__ ├── metrics.cpython-38.pyc └── util.cpython-38.pyc ├── metrics.py └── util.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/LICENSE -------------------------------------------------------------------------------- /Params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/Params.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/README.md -------------------------------------------------------------------------------- /componenets/__pycache__/attention.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/attention.cpython-38.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/engine.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/engine.cpython-38.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/metrics.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/metrics.cpython-310.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/metrics.cpython-38.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/metrics.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/metrics.cpython-39.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/new_metrics.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/new_metrics.cpython-310.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/new_metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/new_metrics.cpython-38.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/new_metrics.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/new_metrics.cpython-39.pyc -------------------------------------------------------------------------------- /componenets/__pycache__/normalization.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/__pycache__/normalization.cpython-38.pyc -------------------------------------------------------------------------------- /componenets/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/metrics.py -------------------------------------------------------------------------------- /componenets/new_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/new_metrics.py -------------------------------------------------------------------------------- /componenets/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/componenets/normalization.py -------------------------------------------------------------------------------- /config/STExplainer_pems4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/config/STExplainer_pems4.yaml -------------------------------------------------------------------------------- /data/DataHandler_st.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/DataHandler_st.py -------------------------------------------------------------------------------- /data/PEMS04/PEMS04.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/PEMS04/PEMS04.csv -------------------------------------------------------------------------------- /data/PEMS04/PEMS04.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/PEMS04/PEMS04.npz -------------------------------------------------------------------------------- /data/__pycache__/DataHandler.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/__pycache__/DataHandler.cpython-38.pyc -------------------------------------------------------------------------------- /data/__pycache__/DataHandler_pyg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/__pycache__/DataHandler_pyg.cpython-38.pyc -------------------------------------------------------------------------------- /data/__pycache__/DataHandler_st.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/__pycache__/DataHandler_st.cpython-38.pyc -------------------------------------------------------------------------------- /data/__pycache__/STGDataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/__pycache__/STGDataset.cpython-38.pyc -------------------------------------------------------------------------------- /data/__pycache__/STGDataset_pyg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/data/__pycache__/STGDataset_pyg.cpython-38.pyc -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/engine.py -------------------------------------------------------------------------------- /images/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/images/.DS_Store -------------------------------------------------------------------------------- /images/STExplainer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/images/STExplainer.png -------------------------------------------------------------------------------- /methods/STExplainer/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/MLP.py -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/MLP.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/MLP.cpython-310.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/MLP.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/MLP.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/MLP.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/MLP.cpython-39.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/STID_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/STID_arch.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/STID_mo.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/STID_mo.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/gat_gib.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/gat_gib.cpython-310.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/gat_gib.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/gat_gib.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/gatconv_gib.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/gatconv_gib.cpython-310.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/gatconv_gib.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/gatconv_gib.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/model_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/model_utils.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/stgib.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/stgib.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/stgib_series.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/stgib_series.cpython-310.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/stgib_series.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/stgib_series.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/__pycache__/stgib_series.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/__pycache__/stgib_series.cpython-39.pyc -------------------------------------------------------------------------------- /methods/STExplainer/gat_gib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/gat_gib.py -------------------------------------------------------------------------------- /methods/STExplainer/gatconv_gib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/gatconv_gib.py -------------------------------------------------------------------------------- /methods/STExplainer/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/model_utils.py -------------------------------------------------------------------------------- /methods/STExplainer/stgat/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/MLP.py -------------------------------------------------------------------------------- /methods/STExplainer/stgat/__pycache__/MLP.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/__pycache__/MLP.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgat/__pycache__/conv_layers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/__pycache__/conv_layers.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgat/__pycache__/gat.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/__pycache__/gat.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgat/__pycache__/stgat.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/__pycache__/stgat.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgat/conv_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/conv_layers.py -------------------------------------------------------------------------------- /methods/STExplainer/stgat/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/gat.py -------------------------------------------------------------------------------- /methods/STExplainer/stgat/stgat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgat/stgat.py -------------------------------------------------------------------------------- /methods/STExplainer/stgib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgib.py -------------------------------------------------------------------------------- /methods/STExplainer/stgib_series.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgib_series.py -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/MLP.py -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/__pycache__/MLP.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/__pycache__/MLP.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/__pycache__/conv_layers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/__pycache__/conv_layers.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/__pycache__/gat_ib.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/__pycache__/gat_ib.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/__pycache__/get_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/__pycache__/get_model.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/__pycache__/gsat.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/__pycache__/gsat.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/__pycache__/stgsat.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/__pycache__/stgsat.cpython-38.pyc -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/conv_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/conv_layers.py -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/gat_ib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/gat_ib.py -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/get_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/get_model.py -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/gsat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/gsat.py -------------------------------------------------------------------------------- /methods/STExplainer/stgsat/stgsat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/methods/STExplainer/stgsat/stgsat.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/model/2023y10m24d/STExplainer_pems4.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/results/model/2023y10m24d/STExplainer_pems4.pkl -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/train.py -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/utils/__pycache__/metrics.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/utils/__pycache__/util.cpython-38.pyc -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/STExplainer/HEAD/utils/util.py --------------------------------------------------------------------------------