This is the official PyTorch implementation of ContrastPool from the paper "Contrasformer: A Brain Network Contrastive Transformer for Neurodegenerative Condition Identification" published in 33rd ACM International Conference on Information and Knowledge Management (CIKM'24).
All Preprocessed data used in this paper are published in this paper.
Data splits and configurations are stored in ./data/
and ./configs/
. If you want to process your own data, please check the dataloader script ./data/BrainNet.py
.
Please check experiment.sh
on how to run the project.
If you have any questions, please feel free to reach out at jiaxing003@e.ntu.edu.sg
.