├── README.md ├── alter ├── __init__.py ├── __main__.py ├── components │ ├── __init__.py │ ├── logger.py │ ├── lr_scheduler.py │ └── optimizer.py ├── conf │ ├── config.yaml │ ├── dataset │ │ └── ABIDE.yaml │ ├── datasz │ │ ├── 100p.yaml │ │ ├── 10p.yaml │ │ ├── 20p.yaml │ │ ├── 30p.yaml │ │ ├── 40p.yaml │ │ ├── 50p.yaml │ │ ├── 60p.yaml │ │ ├── 70p.yaml │ │ ├── 80p.yaml │ │ └── 90p.yaml │ ├── model │ │ ├── brainnetcnn.yaml │ │ ├── fbnetgen.yaml │ │ ├── lrbgt.yaml │ │ └── transformer.yaml │ ├── optimizer │ │ └── adam.yaml │ ├── preprocess │ │ ├── mixup.yaml │ │ └── non_mixup.yaml │ └── training │ │ └── basic_training.yaml ├── dataset │ ├── __init__.py │ ├── abide.py │ ├── dataloader.py │ └── preprocess.py ├── models │ ├── LRBGT │ │ ├── __init__.py │ │ ├── components │ │ │ ├── __init__.py │ │ │ └── transformer_encoder.py │ │ ├── lrbgt.py │ │ └── ptdec │ │ │ ├── __init__.py │ │ │ ├── cluster.py │ │ │ └── dec.py │ ├── __init__.py │ ├── base.py │ ├── brainnetcnn.py │ └── transformer.py ├── training │ ├── __init__.py │ └── training.py └── utils │ ├── __init__.py │ ├── accuracy.py │ ├── count_params.py │ ├── meter.py │ └── prepossess.py └── figure └── figure1.jpg /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/README.md -------------------------------------------------------------------------------- /alter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /alter/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/__main__.py -------------------------------------------------------------------------------- /alter/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/components/__init__.py -------------------------------------------------------------------------------- /alter/components/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/components/logger.py -------------------------------------------------------------------------------- /alter/components/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/components/lr_scheduler.py -------------------------------------------------------------------------------- /alter/components/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/components/optimizer.py -------------------------------------------------------------------------------- /alter/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/conf/config.yaml -------------------------------------------------------------------------------- /alter/conf/dataset/ABIDE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/conf/dataset/ABIDE.yaml -------------------------------------------------------------------------------- /alter/conf/datasz/100p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 1. 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/10p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.1 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/20p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.2 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/30p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.3 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/40p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.4 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/50p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.5 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/60p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.6 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/70p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.7 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/80p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.8 2 | -------------------------------------------------------------------------------- /alter/conf/datasz/90p.yaml: -------------------------------------------------------------------------------- 1 | percentage: 0.9 2 | -------------------------------------------------------------------------------- /alter/conf/model/brainnetcnn.yaml: -------------------------------------------------------------------------------- 1 | name: BrainNetCNN 2 | -------------------------------------------------------------------------------- /alter/conf/model/fbnetgen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/conf/model/fbnetgen.yaml -------------------------------------------------------------------------------- /alter/conf/model/lrbgt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/conf/model/lrbgt.yaml -------------------------------------------------------------------------------- /alter/conf/model/transformer.yaml: -------------------------------------------------------------------------------- 1 | name: GraphTransformer 2 | self_attention_layer: 2 3 | readout: concat 4 | -------------------------------------------------------------------------------- /alter/conf/optimizer/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/conf/optimizer/adam.yaml -------------------------------------------------------------------------------- /alter/conf/preprocess/mixup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/conf/preprocess/mixup.yaml -------------------------------------------------------------------------------- /alter/conf/preprocess/non_mixup.yaml: -------------------------------------------------------------------------------- 1 | name: non_mixup 2 | continus: False -------------------------------------------------------------------------------- /alter/conf/training/basic_training.yaml: -------------------------------------------------------------------------------- 1 | name: Train 2 | epochs: 200 -------------------------------------------------------------------------------- /alter/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/dataset/__init__.py -------------------------------------------------------------------------------- /alter/dataset/abide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/dataset/abide.py -------------------------------------------------------------------------------- /alter/dataset/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/dataset/dataloader.py -------------------------------------------------------------------------------- /alter/dataset/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/dataset/preprocess.py -------------------------------------------------------------------------------- /alter/models/LRBGT/__init__.py: -------------------------------------------------------------------------------- 1 | from .lrbgt import BrainNetworkTransformer 2 | -------------------------------------------------------------------------------- /alter/models/LRBGT/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/LRBGT/components/__init__.py -------------------------------------------------------------------------------- /alter/models/LRBGT/components/transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/LRBGT/components/transformer_encoder.py -------------------------------------------------------------------------------- /alter/models/LRBGT/lrbgt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/LRBGT/lrbgt.py -------------------------------------------------------------------------------- /alter/models/LRBGT/ptdec/__init__.py: -------------------------------------------------------------------------------- 1 | from .dec import DEC 2 | -------------------------------------------------------------------------------- /alter/models/LRBGT/ptdec/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/LRBGT/ptdec/cluster.py -------------------------------------------------------------------------------- /alter/models/LRBGT/ptdec/dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/LRBGT/ptdec/dec.py -------------------------------------------------------------------------------- /alter/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/__init__.py -------------------------------------------------------------------------------- /alter/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/base.py -------------------------------------------------------------------------------- /alter/models/brainnetcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/brainnetcnn.py -------------------------------------------------------------------------------- /alter/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/models/transformer.py -------------------------------------------------------------------------------- /alter/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/training/__init__.py -------------------------------------------------------------------------------- /alter/training/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/training/training.py -------------------------------------------------------------------------------- /alter/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/utils/__init__.py -------------------------------------------------------------------------------- /alter/utils/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/utils/accuracy.py -------------------------------------------------------------------------------- /alter/utils/count_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/utils/count_params.py -------------------------------------------------------------------------------- /alter/utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/utils/meter.py -------------------------------------------------------------------------------- /alter/utils/prepossess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/alter/utils/prepossess.py -------------------------------------------------------------------------------- /figure/figure1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yushuowiki/ALTER/HEAD/figure/figure1.jpg --------------------------------------------------------------------------------