├── .gitignore ├── LICENSE ├── README.md ├── data └── BA3 │ └── raw │ └── BA-3motif.npy ├── datasets ├── __init__.py ├── ba3motif_dataset.py ├── graphss2_dataset.py ├── mutag_dataset.py └── vg_dataset.py ├── evaluate.py ├── example.py ├── example_visual.ipynb ├── explainers ├── __init__.py ├── base.py ├── common.py ├── cxplainer.py ├── deeplift.py ├── gnnexplainer.py ├── gradcam.py ├── ig_explainer.py ├── meta_gnnexplainer.py ├── meta_pgm_explainer.py ├── pg_explainer.py ├── pgm_explainer.py ├── random_caster.py ├── refine.py ├── sa_explainer.py ├── screener.py └── visual.py ├── framework.png ├── gnns ├── __init__.py ├── ba3motif_gnn.py ├── mnist_gnn.py ├── mutag_gnn.py ├── overloader.py ├── run.ssh └── vg_gnn.py ├── log ├── pg-ba3.log ├── pg-mnist.log ├── pg-mutag.log ├── pg-vg.log ├── refine-ba3.log ├── refine-mnist.log ├── refine-mutag.log └── refine-vg.log ├── param └── gnns │ ├── ba3_net.pt │ ├── mnist_net.pt │ ├── mutag_net.pt │ └── vg_net.pt ├── results ├── ba3_results.json ├── mnist_results.json ├── mutag_results.json └── vg_results.json ├── train ├── __init__.py ├── config.py ├── pg_train.py ├── refine_train.py └── run.sh └── utils ├── __init__.py ├── dataset.py ├── get_subgraph.py ├── helper.py ├── logger.py ├── parser.py ├── saver.py └── train_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/README.md -------------------------------------------------------------------------------- /data/BA3/raw/BA-3motif.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/data/BA3/raw/BA-3motif.npy -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/ba3motif_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/datasets/ba3motif_dataset.py -------------------------------------------------------------------------------- /datasets/graphss2_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/datasets/graphss2_dataset.py -------------------------------------------------------------------------------- /datasets/mutag_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/datasets/mutag_dataset.py -------------------------------------------------------------------------------- /datasets/vg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/datasets/vg_dataset.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/evaluate.py -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/example.py -------------------------------------------------------------------------------- /example_visual.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/example_visual.ipynb -------------------------------------------------------------------------------- /explainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/__init__.py -------------------------------------------------------------------------------- /explainers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/base.py -------------------------------------------------------------------------------- /explainers/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/common.py -------------------------------------------------------------------------------- /explainers/cxplainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/cxplainer.py -------------------------------------------------------------------------------- /explainers/deeplift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/deeplift.py -------------------------------------------------------------------------------- /explainers/gnnexplainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/gnnexplainer.py -------------------------------------------------------------------------------- /explainers/gradcam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/gradcam.py -------------------------------------------------------------------------------- /explainers/ig_explainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/ig_explainer.py -------------------------------------------------------------------------------- /explainers/meta_gnnexplainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/meta_gnnexplainer.py -------------------------------------------------------------------------------- /explainers/meta_pgm_explainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/meta_pgm_explainer.py -------------------------------------------------------------------------------- /explainers/pg_explainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/pg_explainer.py -------------------------------------------------------------------------------- /explainers/pgm_explainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/pgm_explainer.py -------------------------------------------------------------------------------- /explainers/random_caster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/random_caster.py -------------------------------------------------------------------------------- /explainers/refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/refine.py -------------------------------------------------------------------------------- /explainers/sa_explainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/sa_explainer.py -------------------------------------------------------------------------------- /explainers/screener.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/screener.py -------------------------------------------------------------------------------- /explainers/visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/explainers/visual.py -------------------------------------------------------------------------------- /framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/framework.png -------------------------------------------------------------------------------- /gnns/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/__init__.py -------------------------------------------------------------------------------- /gnns/ba3motif_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/ba3motif_gnn.py -------------------------------------------------------------------------------- /gnns/mnist_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/mnist_gnn.py -------------------------------------------------------------------------------- /gnns/mutag_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/mutag_gnn.py -------------------------------------------------------------------------------- /gnns/overloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/overloader.py -------------------------------------------------------------------------------- /gnns/run.ssh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/run.ssh -------------------------------------------------------------------------------- /gnns/vg_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/gnns/vg_gnn.py -------------------------------------------------------------------------------- /log/pg-ba3.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/pg-ba3.log -------------------------------------------------------------------------------- /log/pg-mnist.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/pg-mnist.log -------------------------------------------------------------------------------- /log/pg-mutag.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/pg-mutag.log -------------------------------------------------------------------------------- /log/pg-vg.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/pg-vg.log -------------------------------------------------------------------------------- /log/refine-ba3.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/refine-ba3.log -------------------------------------------------------------------------------- /log/refine-mnist.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/refine-mnist.log -------------------------------------------------------------------------------- /log/refine-mutag.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/refine-mutag.log -------------------------------------------------------------------------------- /log/refine-vg.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/log/refine-vg.log -------------------------------------------------------------------------------- /param/gnns/ba3_net.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/param/gnns/ba3_net.pt -------------------------------------------------------------------------------- /param/gnns/mnist_net.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/param/gnns/mnist_net.pt -------------------------------------------------------------------------------- /param/gnns/mutag_net.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/param/gnns/mutag_net.pt -------------------------------------------------------------------------------- /param/gnns/vg_net.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/param/gnns/vg_net.pt -------------------------------------------------------------------------------- /results/ba3_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/results/ba3_results.json -------------------------------------------------------------------------------- /results/mnist_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/results/mnist_results.json -------------------------------------------------------------------------------- /results/mutag_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/results/mutag_results.json -------------------------------------------------------------------------------- /results/vg_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/results/vg_results.json -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/train/config.py -------------------------------------------------------------------------------- /train/pg_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/train/pg_train.py -------------------------------------------------------------------------------- /train/refine_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/train/refine_train.py -------------------------------------------------------------------------------- /train/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/train/run.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/get_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/get_subgraph.py -------------------------------------------------------------------------------- /utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/helper.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/saver.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Wuyxin/ReFine/HEAD/utils/train_utils.py --------------------------------------------------------------------------------