├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── baseline.sh ├── configs ├── abide_schaefer100 │ └── TUs_graph_classification_ContrastPool_abide_schaefer100_100k.json ├── adni_schaefer100 │ └── TUs_graph_classification_ContrastPool_adni_schaefer100_100k.json ├── neurocon_schaefer100 │ └── TUs_graph_classification_ContrastPool_neurocon_schaefer100_100k.json ├── ppmi_schaefer100 │ └── TUs_graph_classification_ContrastPool_ppmi_schaefer100_100k.json └── taowu_schaefer100 │ └── TUs_graph_classification_ContrastPool_taowu_schaefer100_100k.json ├── contrast_subgraph.py ├── data ├── BrainNet.py ├── abide_schaefer100 │ ├── test.index │ ├── train.index │ └── val.index ├── adni_schaefer100 │ ├── test.index │ ├── train.index │ └── val.index ├── data.py ├── generate_data_from_mat.py ├── neurocon_schaefer100 │ ├── test.index │ ├── train.index │ └── val.index ├── ppmi_schaefer100 │ ├── test.index │ ├── train.index │ └── val.index └── taowu_schaefer100 │ ├── test.index │ ├── train.index │ └── val.index ├── figs └── framework.png ├── layers ├── attention_layer.py ├── contrastpool_layer.py ├── diffpool_layer.py └── graphsage_layer.py ├── main.py ├── metrics.py ├── nets ├── contrastpool_net.py └── load_net.py └── train_TUs_graph_classification.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/README.md -------------------------------------------------------------------------------- /baseline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/baseline.sh -------------------------------------------------------------------------------- /configs/abide_schaefer100/TUs_graph_classification_ContrastPool_abide_schaefer100_100k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/configs/abide_schaefer100/TUs_graph_classification_ContrastPool_abide_schaefer100_100k.json -------------------------------------------------------------------------------- /configs/adni_schaefer100/TUs_graph_classification_ContrastPool_adni_schaefer100_100k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/configs/adni_schaefer100/TUs_graph_classification_ContrastPool_adni_schaefer100_100k.json -------------------------------------------------------------------------------- /configs/neurocon_schaefer100/TUs_graph_classification_ContrastPool_neurocon_schaefer100_100k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/configs/neurocon_schaefer100/TUs_graph_classification_ContrastPool_neurocon_schaefer100_100k.json -------------------------------------------------------------------------------- /configs/ppmi_schaefer100/TUs_graph_classification_ContrastPool_ppmi_schaefer100_100k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/configs/ppmi_schaefer100/TUs_graph_classification_ContrastPool_ppmi_schaefer100_100k.json -------------------------------------------------------------------------------- /configs/taowu_schaefer100/TUs_graph_classification_ContrastPool_taowu_schaefer100_100k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/configs/taowu_schaefer100/TUs_graph_classification_ContrastPool_taowu_schaefer100_100k.json -------------------------------------------------------------------------------- /contrast_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/contrast_subgraph.py -------------------------------------------------------------------------------- /data/BrainNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/BrainNet.py -------------------------------------------------------------------------------- /data/abide_schaefer100/test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/abide_schaefer100/test.index -------------------------------------------------------------------------------- /data/abide_schaefer100/train.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/abide_schaefer100/train.index -------------------------------------------------------------------------------- /data/abide_schaefer100/val.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/abide_schaefer100/val.index -------------------------------------------------------------------------------- /data/adni_schaefer100/test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/adni_schaefer100/test.index -------------------------------------------------------------------------------- /data/adni_schaefer100/train.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/adni_schaefer100/train.index -------------------------------------------------------------------------------- /data/adni_schaefer100/val.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/adni_schaefer100/val.index -------------------------------------------------------------------------------- /data/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/data.py -------------------------------------------------------------------------------- /data/generate_data_from_mat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/generate_data_from_mat.py -------------------------------------------------------------------------------- /data/neurocon_schaefer100/test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/neurocon_schaefer100/test.index -------------------------------------------------------------------------------- /data/neurocon_schaefer100/train.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/neurocon_schaefer100/train.index -------------------------------------------------------------------------------- /data/neurocon_schaefer100/val.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/neurocon_schaefer100/val.index -------------------------------------------------------------------------------- /data/ppmi_schaefer100/test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/ppmi_schaefer100/test.index -------------------------------------------------------------------------------- /data/ppmi_schaefer100/train.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/ppmi_schaefer100/train.index -------------------------------------------------------------------------------- /data/ppmi_schaefer100/val.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/ppmi_schaefer100/val.index -------------------------------------------------------------------------------- /data/taowu_schaefer100/test.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/taowu_schaefer100/test.index -------------------------------------------------------------------------------- /data/taowu_schaefer100/train.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/taowu_schaefer100/train.index -------------------------------------------------------------------------------- /data/taowu_schaefer100/val.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/data/taowu_schaefer100/val.index -------------------------------------------------------------------------------- /figs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/figs/framework.png -------------------------------------------------------------------------------- /layers/attention_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/layers/attention_layer.py -------------------------------------------------------------------------------- /layers/contrastpool_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/layers/contrastpool_layer.py -------------------------------------------------------------------------------- /layers/diffpool_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/layers/diffpool_layer.py -------------------------------------------------------------------------------- /layers/graphsage_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/layers/graphsage_layer.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/main.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/metrics.py -------------------------------------------------------------------------------- /nets/contrastpool_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/nets/contrastpool_net.py -------------------------------------------------------------------------------- /nets/load_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/nets/load_net.py -------------------------------------------------------------------------------- /train_TUs_graph_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AngusMonroe/ContrastPool/HEAD/train_TUs_graph_classification.py --------------------------------------------------------------------------------