├── BGRL ├── bgrl │ ├── __init__.py │ ├── bgrl.py │ ├── data.py │ ├── get_args.py │ ├── models.py │ ├── predictors.py │ ├── scheduler.py │ ├── transforms.py │ ├── transforms_new.py │ └── utils.py ├── new_struct.py ├── run_GAugLLM.sh ├── run_original.sh └── train_transductive.py ├── GBT ├── GCL │ ├── __init__.py │ ├── augmentors │ │ ├── __init__.py │ │ ├── augmentor.py │ │ ├── edge_removing.py │ │ ├── feature_masking.py │ │ ├── functional.py │ │ ├── identity.py │ │ └── ppr_diffusion.py │ ├── losses │ │ ├── __init__.py │ │ ├── barlow_twins.py │ │ ├── bootstrap.py │ │ ├── infonce.py │ │ ├── jsd.py │ │ └── losses.py │ ├── models │ │ ├── __init__.py │ │ ├── contrast_model.py │ │ └── samplers.py │ └── utils.py ├── experiments │ ├── configs │ │ └── train_transductive.yaml │ └── scripts │ │ ├── run_GAugGBT.sh │ │ ├── run_GAugGraphCL.sh │ │ ├── run_ori.sh │ │ ├── train_transductive.py │ │ └── train_transductive_struct.py └── gssl │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── datasets.cpython-310.pyc │ ├── discriminator.cpython-310.pyc │ ├── loss.cpython-310.pyc │ ├── readout.cpython-310.pyc │ ├── tasks.cpython-310.pyc │ ├── transductive_model.cpython-310.pyc │ ├── transductive_model_GBT.cpython-310.pyc │ ├── transductive_model_GraphCL.cpython-310.pyc │ ├── transductive_model_arxiv.cpython-310.pyc │ ├── transductive_model_arxiv_GBT.cpython-310.pyc │ ├── transductive_model_arxiv_GraphCL.cpython-310.pyc │ ├── transductive_model_arxiv_ori.cpython-310.pyc │ ├── transductive_model_gpt.cpython-310.pyc │ ├── transductive_model_ori.cpython-310.pyc │ └── utils.cpython-310.pyc │ ├── augment.py │ ├── datasets.py │ ├── discriminator.py │ ├── loss.py │ ├── readout.py │ ├── tasks.py │ ├── transductive_model.py │ ├── transductive_model_GBT.py │ ├── transductive_model_GraphCL.py │ ├── transductive_model_arxiv_GBT.py │ ├── transductive_model_arxiv_GraphCL.py │ ├── transductive_model_newStruct.py │ ├── transductive_model_ori.py │ └── utils.py ├── GNNs ├── GCN.py ├── GNNs.py ├── GNNs_struct.py ├── run_ori.py └── run_struct.py ├── GraphMAE ├── chem │ ├── README.md │ ├── batch.py │ ├── dataloader.py │ ├── finetune.py │ ├── finetune.sh │ ├── init_weights │ │ └── 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 ├── main_transductive.py ├── run.sh ├── scripts │ └── run_transductive.sh └── utils.py ├── LLMs ├── mix-of-expert-prompt │ ├── GiantTrainer.py │ ├── GiantTrainer_new.py │ ├── arxivMatcher.py │ ├── arxivModel.py │ ├── arxivTrainer.py │ ├── historyMatcher.py │ ├── historyModel.py │ ├── historyTrainer.py │ ├── matcher.py │ ├── model.py │ ├── modelori.py │ ├── pubmedMatcher.py │ ├── pubmedModel.py │ ├── pubmedTrainer.py │ ├── trainModel.py │ └── utils.py └── run.sh ├── MIT-LICENSE ├── README.md ├── S2GAE ├── logger.py ├── model.py ├── run.sh ├── s2gae_nc_acc.py └── utils.py ├── data_utils ├── dataset.py ├── load.py ├── load_arxiv.py ├── load_cora.py ├── load_pubmed.py ├── logistic_regression_eval.py └── ogbn_arxiv_split_idx.json └── img ├── moep.png └── pipeline.png /BGRL/bgrl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/__init__.py -------------------------------------------------------------------------------- /BGRL/bgrl/bgrl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/bgrl.py -------------------------------------------------------------------------------- /BGRL/bgrl/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/data.py -------------------------------------------------------------------------------- /BGRL/bgrl/get_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/get_args.py -------------------------------------------------------------------------------- /BGRL/bgrl/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/models.py -------------------------------------------------------------------------------- /BGRL/bgrl/predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/predictors.py -------------------------------------------------------------------------------- /BGRL/bgrl/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/scheduler.py -------------------------------------------------------------------------------- /BGRL/bgrl/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/transforms.py -------------------------------------------------------------------------------- /BGRL/bgrl/transforms_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/transforms_new.py -------------------------------------------------------------------------------- /BGRL/bgrl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/bgrl/utils.py -------------------------------------------------------------------------------- /BGRL/new_struct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/new_struct.py -------------------------------------------------------------------------------- /BGRL/run_GAugLLM.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/run_GAugLLM.sh -------------------------------------------------------------------------------- /BGRL/run_original.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/run_original.sh -------------------------------------------------------------------------------- /BGRL/train_transductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/BGRL/train_transductive.py -------------------------------------------------------------------------------- /GBT/GCL/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /GBT/GCL/augmentors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/__init__.py -------------------------------------------------------------------------------- /GBT/GCL/augmentors/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/augmentor.py -------------------------------------------------------------------------------- /GBT/GCL/augmentors/edge_removing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/edge_removing.py -------------------------------------------------------------------------------- /GBT/GCL/augmentors/feature_masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/feature_masking.py -------------------------------------------------------------------------------- /GBT/GCL/augmentors/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/functional.py -------------------------------------------------------------------------------- /GBT/GCL/augmentors/identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/identity.py -------------------------------------------------------------------------------- /GBT/GCL/augmentors/ppr_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/augmentors/ppr_diffusion.py -------------------------------------------------------------------------------- /GBT/GCL/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/losses/__init__.py -------------------------------------------------------------------------------- /GBT/GCL/losses/barlow_twins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/losses/barlow_twins.py -------------------------------------------------------------------------------- /GBT/GCL/losses/bootstrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/losses/bootstrap.py -------------------------------------------------------------------------------- /GBT/GCL/losses/infonce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/losses/infonce.py -------------------------------------------------------------------------------- /GBT/GCL/losses/jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/losses/jsd.py -------------------------------------------------------------------------------- /GBT/GCL/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/losses/losses.py -------------------------------------------------------------------------------- /GBT/GCL/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/models/__init__.py -------------------------------------------------------------------------------- /GBT/GCL/models/contrast_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/models/contrast_model.py -------------------------------------------------------------------------------- /GBT/GCL/models/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/models/samplers.py -------------------------------------------------------------------------------- /GBT/GCL/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/GCL/utils.py -------------------------------------------------------------------------------- /GBT/experiments/configs/train_transductive.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/experiments/configs/train_transductive.yaml -------------------------------------------------------------------------------- /GBT/experiments/scripts/run_GAugGBT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/experiments/scripts/run_GAugGBT.sh -------------------------------------------------------------------------------- /GBT/experiments/scripts/run_GAugGraphCL.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/experiments/scripts/run_GAugGraphCL.sh -------------------------------------------------------------------------------- /GBT/experiments/scripts/run_ori.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/experiments/scripts/run_ori.sh -------------------------------------------------------------------------------- /GBT/experiments/scripts/train_transductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/experiments/scripts/train_transductive.py -------------------------------------------------------------------------------- /GBT/experiments/scripts/train_transductive_struct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/experiments/scripts/train_transductive_struct.py -------------------------------------------------------------------------------- /GBT/gssl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__init__.py -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/datasets.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/datasets.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/discriminator.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/discriminator.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/loss.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/loss.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/readout.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/readout.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/tasks.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/tasks.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_GBT.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_GBT.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_GraphCL.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_GraphCL.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_arxiv.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_arxiv.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_arxiv_GBT.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_arxiv_GBT.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_arxiv_GraphCL.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_arxiv_GraphCL.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_arxiv_ori.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_arxiv_ori.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_gpt.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_gpt.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/transductive_model_ori.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/transductive_model_ori.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/__pycache__/utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/__pycache__/utils.cpython-310.pyc -------------------------------------------------------------------------------- /GBT/gssl/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/augment.py -------------------------------------------------------------------------------- /GBT/gssl/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/datasets.py -------------------------------------------------------------------------------- /GBT/gssl/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/discriminator.py -------------------------------------------------------------------------------- /GBT/gssl/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/loss.py -------------------------------------------------------------------------------- /GBT/gssl/readout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/readout.py -------------------------------------------------------------------------------- /GBT/gssl/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/tasks.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model_GBT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model_GBT.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model_GraphCL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model_GraphCL.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model_arxiv_GBT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model_arxiv_GBT.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model_arxiv_GraphCL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model_arxiv_GraphCL.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model_newStruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model_newStruct.py -------------------------------------------------------------------------------- /GBT/gssl/transductive_model_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/transductive_model_ori.py -------------------------------------------------------------------------------- /GBT/gssl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GBT/gssl/utils.py -------------------------------------------------------------------------------- /GNNs/GCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GNNs/GCN.py -------------------------------------------------------------------------------- /GNNs/GNNs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GNNs/GNNs.py -------------------------------------------------------------------------------- /GNNs/GNNs_struct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GNNs/GNNs_struct.py -------------------------------------------------------------------------------- /GNNs/run_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GNNs/run_ori.py -------------------------------------------------------------------------------- /GNNs/run_struct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GNNs/run_struct.py -------------------------------------------------------------------------------- /GraphMAE/chem/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/README.md -------------------------------------------------------------------------------- /GraphMAE/chem/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/batch.py -------------------------------------------------------------------------------- /GraphMAE/chem/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/dataloader.py -------------------------------------------------------------------------------- /GraphMAE/chem/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/finetune.py -------------------------------------------------------------------------------- /GraphMAE/chem/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/finetune.sh -------------------------------------------------------------------------------- /GraphMAE/chem/init_weights/pretrained.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/init_weights/pretrained.pth -------------------------------------------------------------------------------- /GraphMAE/chem/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/loader.py -------------------------------------------------------------------------------- /GraphMAE/chem/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/model.py -------------------------------------------------------------------------------- /GraphMAE/chem/parse_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/parse_result.py -------------------------------------------------------------------------------- /GraphMAE/chem/pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/pretraining.py -------------------------------------------------------------------------------- /GraphMAE/chem/splitters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/splitters.py -------------------------------------------------------------------------------- /GraphMAE/chem/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/chem/util.py -------------------------------------------------------------------------------- /GraphMAE/configs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/configs.yml -------------------------------------------------------------------------------- /GraphMAE/graphmae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /GraphMAE/graphmae/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /GraphMAE/graphmae/datasets/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/datasets/data_util.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/evaluation.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/__init__.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/dot_gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/dot_gat.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/edcoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/edcoder.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/gat.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/gcn.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/gin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/gin.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/models/loss_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/models/loss_func.py -------------------------------------------------------------------------------- /GraphMAE/graphmae/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/graphmae/utils.py -------------------------------------------------------------------------------- /GraphMAE/main_transductive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/main_transductive.py -------------------------------------------------------------------------------- /GraphMAE/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/run.sh -------------------------------------------------------------------------------- /GraphMAE/scripts/run_transductive.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/scripts/run_transductive.sh -------------------------------------------------------------------------------- /GraphMAE/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/GraphMAE/utils.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/GiantTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/GiantTrainer.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/GiantTrainer_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/GiantTrainer_new.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/arxivMatcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/arxivMatcher.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/arxivModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/arxivModel.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/arxivTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/arxivTrainer.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/historyMatcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/historyMatcher.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/historyModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/historyModel.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/historyTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/historyTrainer.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/matcher.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/model.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/modelori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/modelori.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/pubmedMatcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/pubmedMatcher.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/pubmedModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/pubmedModel.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/pubmedTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/pubmedTrainer.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/trainModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/trainModel.py -------------------------------------------------------------------------------- /LLMs/mix-of-expert-prompt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/mix-of-expert-prompt/utils.py -------------------------------------------------------------------------------- /LLMs/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/LLMs/run.sh -------------------------------------------------------------------------------- /MIT-LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/MIT-LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/README.md -------------------------------------------------------------------------------- /S2GAE/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/S2GAE/logger.py -------------------------------------------------------------------------------- /S2GAE/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/S2GAE/model.py -------------------------------------------------------------------------------- /S2GAE/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/S2GAE/run.sh -------------------------------------------------------------------------------- /S2GAE/s2gae_nc_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/S2GAE/s2gae_nc_acc.py -------------------------------------------------------------------------------- /S2GAE/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/S2GAE/utils.py -------------------------------------------------------------------------------- /data_utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/dataset.py -------------------------------------------------------------------------------- /data_utils/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/load.py -------------------------------------------------------------------------------- /data_utils/load_arxiv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/load_arxiv.py -------------------------------------------------------------------------------- /data_utils/load_cora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/load_cora.py -------------------------------------------------------------------------------- /data_utils/load_pubmed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/load_pubmed.py -------------------------------------------------------------------------------- /data_utils/logistic_regression_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/logistic_regression_eval.py -------------------------------------------------------------------------------- /data_utils/ogbn_arxiv_split_idx.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/data_utils/ogbn_arxiv_split_idx.json -------------------------------------------------------------------------------- /img/moep.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/img/moep.png -------------------------------------------------------------------------------- /img/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NYUSHCS/GAugLLM/HEAD/img/pipeline.png --------------------------------------------------------------------------------