├── .gitignore ├── LICENSE ├── README.md ├── chem ├── .DS_Store ├── README.md ├── batch.py ├── dataloader.py ├── finetune.py ├── finetune.sh ├── init_weights │ ├── .DS_Store │ └── pretrained.pth ├── loader.py ├── model.py ├── parse_result.py ├── pretraining.py ├── splitters.py └── util.py ├── configs.yml ├── graphmae ├── __init__.py ├── datasets │ ├── __init__.py │ └── data_util.py ├── evaluation.py ├── models │ ├── __init__.py │ ├── dot_gat.py │ ├── edcoder.py │ ├── gat.py │ ├── gcn.py │ ├── gin.py │ └── loss_func.py └── utils.py ├── imgs ├── .DS_Store ├── ablation.jpg ├── compare.png └── fig.png ├── main_graph.py ├── main_inductive.py ├── main_transductive.py ├── requirements.txt └── scripts ├── run_graph.sh ├── run_inductive.sh └── run_transductive.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/README.md -------------------------------------------------------------------------------- /chem/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/.DS_Store -------------------------------------------------------------------------------- /chem/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/README.md -------------------------------------------------------------------------------- /chem/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/batch.py -------------------------------------------------------------------------------- /chem/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/dataloader.py -------------------------------------------------------------------------------- /chem/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/finetune.py -------------------------------------------------------------------------------- /chem/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/finetune.sh -------------------------------------------------------------------------------- /chem/init_weights/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/init_weights/.DS_Store -------------------------------------------------------------------------------- /chem/init_weights/pretrained.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/init_weights/pretrained.pth -------------------------------------------------------------------------------- /chem/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/loader.py -------------------------------------------------------------------------------- /chem/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/model.py -------------------------------------------------------------------------------- /chem/parse_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/parse_result.py -------------------------------------------------------------------------------- /chem/pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/pretraining.py -------------------------------------------------------------------------------- /chem/splitters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/splitters.py -------------------------------------------------------------------------------- /chem/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/chem/util.py -------------------------------------------------------------------------------- /configs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/configs.yml -------------------------------------------------------------------------------- /graphmae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /graphmae/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /graphmae/datasets/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/datasets/data_util.py -------------------------------------------------------------------------------- /graphmae/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/evaluation.py -------------------------------------------------------------------------------- /graphmae/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/__init__.py -------------------------------------------------------------------------------- /graphmae/models/dot_gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/dot_gat.py -------------------------------------------------------------------------------- /graphmae/models/edcoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/edcoder.py -------------------------------------------------------------------------------- /graphmae/models/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/gat.py -------------------------------------------------------------------------------- /graphmae/models/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/gcn.py -------------------------------------------------------------------------------- /graphmae/models/gin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/gin.py -------------------------------------------------------------------------------- /graphmae/models/loss_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/models/loss_func.py -------------------------------------------------------------------------------- /graphmae/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/graphmae/utils.py -------------------------------------------------------------------------------- /imgs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/imgs/.DS_Store -------------------------------------------------------------------------------- /imgs/ablation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/imgs/ablation.jpg -------------------------------------------------------------------------------- /imgs/compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/imgs/compare.png -------------------------------------------------------------------------------- /imgs/fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/imgs/fig.png -------------------------------------------------------------------------------- /main_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/main_graph.py -------------------------------------------------------------------------------- /main_inductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/main_inductive.py -------------------------------------------------------------------------------- /main_transductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/main_transductive.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pyyaml 2 | tqdm 3 | tensorboardX 4 | scikit-learn 5 | ogb -------------------------------------------------------------------------------- /scripts/run_graph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/scripts/run_graph.sh -------------------------------------------------------------------------------- /scripts/run_inductive.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/scripts/run_inductive.sh -------------------------------------------------------------------------------- /scripts/run_transductive.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/GraphMAE/HEAD/scripts/run_transductive.sh --------------------------------------------------------------------------------