A deep learning model for data-driven discovery of functional connectivity

https://doi.org/10.3390/a14030075

Usman Mahmood, Zengin Fu, Vince D. Calhoun, Sergey M. Plis

Dependencies:

• PyTorch
• Scikit-Learn
• torch-geometric

conda install pytorch torchvision -c pytorch
conda install sklearn

Installation

Refer to

https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

for torch-geometric installation

# PyTorch
conda install pytorch torchvision -c pytorch
git clone https://github.com/UsmanMahmood27/BrainGNN.git
cd BrainGNN
pip install -e .
pip install -r requirements.txt
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
pip install torch-geometric