# BrainGNN_Pytorch

**Repository Path**: lu_xiang_jiang/BrainGNN_Pytorch

## Basic Information

- **Project Name**: BrainGNN_Pytorch
- **Description**: 论文BrainGNN: Interpretable Brain Graph Neural Network for fMRI Analysis中的模型BrainGNN
论文地址https://www.sciencedirect.com/science/article/pii/S1361841521002784?via%3Dihub
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-10-24
- **Last Updated**: 2021-10-24

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Graph Neural Network for Brain Network Analysis
 A preliminary implementation of BrainGNN


## Usage
### Setup
**pip**

See the `requirements.txt` for environment configuration. 
```bash
pip install -r requirements.txt
```
**PYG**

To install pyg library, [please refer to the document](https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html)

### Dataset 
**ABIDE**

We treat each fMRI as a brain graph. How to download and construct the graphs?
```
python 01-fetch_data.py
python 02-process_data.py
```

### How to run classification?
Training and testing are integrated in file `main.py`. To run
```
python 03-main.py 
```


## Citation
If you find the code and dataset useful, please cite our paper.
```latex
@article{li2020braingnn,
  title={Braingnn: Interpretable brain graph neural network for fmri analysis},
  author={Li, Xiaoxiao and Zhou,Yuan and Dvornek, Nicha and Zhang, Muhan and Gao, Siyuan and Zhuang, Juntang and Scheinost, Dustin and Staib, Lawrence and Ventola, Pamela and Duncan, James},
  journal={bioRxiv},
  year={2020},
  publisher={Cold Spring Harbor Laboratory}
}
```