├── .gitignore ├── LLM4Graph ├── __init__.py ├── configs │ ├── __init__.py │ ├── config.py │ └── tuning.py ├── data │ ├── LinkTAGDataset.py │ ├── OGBNodeTAG.py │ ├── PlanetoidTAG.py │ ├── Pygdata.py │ ├── WikiCSTAG.py │ ├── __init__.py │ ├── lang │ │ ├── __init__.py │ │ └── pe.py │ ├── prompts │ │ ├── __init__.py │ │ └── prompts.py │ ├── transforms.py │ └── utils.py ├── local_dirty_preprocess │ └── preprocess.ipynb ├── models │ ├── __init__.py │ ├── elements │ │ ├── FeatureEncoder.py │ │ ├── MLP.py │ │ ├── __init__.py │ │ ├── gMHA_wrapper.py │ │ ├── gnn.py │ │ └── gnn_wrapper.py │ ├── get_model.py │ ├── gnn │ │ ├── __init__.py │ │ ├── gcn.py │ │ └── gine.py │ ├── graphTrans │ │ ├── ANSGT.py │ │ ├── EGT.py │ │ ├── Exphormer.py │ │ ├── GDWL.py │ │ ├── GKAT.py │ │ ├── GOAT.py │ │ ├── GRIT.py │ │ ├── GraphGPS.py │ │ ├── GraphViT.py │ │ ├── Graphormer.py │ │ ├── NAGphormer.py │ │ ├── RT.py │ │ ├── SAT.py │ │ ├── SGFormer.py │ │ ├── Specformer.py │ │ ├── TokenGT.py │ │ └── __init__.py │ ├── lm │ │ ├── __init__.py │ │ ├── e5.py │ │ ├── llama.py │ │ ├── mistral7b.py │ │ └── sbert.py │ └── misc │ │ └── __init__.py ├── train │ ├── __init__.py │ ├── distributed_llm.py │ ├── eval.py │ ├── full_batch.py │ ├── single_gt.py │ └── utils.py └── utils │ ├── __init__.py │ ├── data_utils.py │ ├── graph.py │ ├── llm │ ├── __init__.py │ ├── api.py │ └── async_utils.py │ ├── misc.py │ ├── normalization.py │ ├── prompt.py │ ├── subgraph.py │ ├── text_utils.py │ └── tuner.py ├── README.md ├── benchmark ├── confs │ └── cora │ │ └── nagphormer.yaml ├── main.py └── sweep.py ├── confs ├── actor │ └── nagphormer.yaml ├── chameleon │ └── nagphormer.yaml ├── citeseer │ └── nagphormer.yaml ├── cora │ └── nagphormer.yaml ├── pokec │ └── nagphormer.yaml ├── products │ └── nagphormer.yaml ├── pubmed │ └── nagphormer.yaml └── squirrel │ └── nagphormer.yaml ├── examples └── LLMs_as_predictors.py ├── main.py └── sweep.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/.gitignore -------------------------------------------------------------------------------- /LLM4Graph/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/configs/config.py -------------------------------------------------------------------------------- /LLM4Graph/configs/tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/configs/tuning.py -------------------------------------------------------------------------------- /LLM4Graph/data/LinkTAGDataset.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/data/OGBNodeTAG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/OGBNodeTAG.py -------------------------------------------------------------------------------- /LLM4Graph/data/PlanetoidTAG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/PlanetoidTAG.py -------------------------------------------------------------------------------- /LLM4Graph/data/Pygdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/Pygdata.py -------------------------------------------------------------------------------- /LLM4Graph/data/WikiCSTAG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/WikiCSTAG.py -------------------------------------------------------------------------------- /LLM4Graph/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/data/lang/__init__.py: -------------------------------------------------------------------------------- 1 | from .pe import * -------------------------------------------------------------------------------- /LLM4Graph/data/lang/pe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/lang/pe.py -------------------------------------------------------------------------------- /LLM4Graph/data/prompts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/data/prompts/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/prompts/prompts.py -------------------------------------------------------------------------------- /LLM4Graph/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/transforms.py -------------------------------------------------------------------------------- /LLM4Graph/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/data/utils.py -------------------------------------------------------------------------------- /LLM4Graph/local_dirty_preprocess/preprocess.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/local_dirty_preprocess/preprocess.ipynb -------------------------------------------------------------------------------- /LLM4Graph/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/__init__.py -------------------------------------------------------------------------------- /LLM4Graph/models/elements/FeatureEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/elements/FeatureEncoder.py -------------------------------------------------------------------------------- /LLM4Graph/models/elements/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/elements/MLP.py -------------------------------------------------------------------------------- /LLM4Graph/models/elements/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/elements/__init__.py -------------------------------------------------------------------------------- /LLM4Graph/models/elements/gMHA_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/elements/gMHA_wrapper.py -------------------------------------------------------------------------------- /LLM4Graph/models/elements/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/elements/gnn.py -------------------------------------------------------------------------------- /LLM4Graph/models/elements/gnn_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/elements/gnn_wrapper.py -------------------------------------------------------------------------------- /LLM4Graph/models/get_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/get_model.py -------------------------------------------------------------------------------- /LLM4Graph/models/gnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/gnn/__init__.py -------------------------------------------------------------------------------- /LLM4Graph/models/gnn/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/gnn/gcn.py -------------------------------------------------------------------------------- /LLM4Graph/models/gnn/gine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/gnn/gine.py -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/ANSGT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/EGT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/Exphormer.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/GDWL.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/GKAT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/GOAT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/GRIT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/GraphGPS.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/GraphViT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/graphTrans/GraphViT.py -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/Graphormer.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/NAGphormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/graphTrans/NAGphormer.py -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/RT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/SAT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/SGFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/graphTrans/SGFormer.py -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/Specformer.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/TokenGT.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/graphTrans/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/graphTrans/__init__.py -------------------------------------------------------------------------------- /LLM4Graph/models/lm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/models/lm/e5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/lm/e5.py -------------------------------------------------------------------------------- /LLM4Graph/models/lm/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/lm/llama.py -------------------------------------------------------------------------------- /LLM4Graph/models/lm/mistral7b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/lm/mistral7b.py -------------------------------------------------------------------------------- /LLM4Graph/models/lm/sbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/models/lm/sbert.py -------------------------------------------------------------------------------- /LLM4Graph/models/misc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/train/__init__.py -------------------------------------------------------------------------------- /LLM4Graph/train/distributed_llm.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/train/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/train/eval.py -------------------------------------------------------------------------------- /LLM4Graph/train/full_batch.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/train/single_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/train/single_gt.py -------------------------------------------------------------------------------- /LLM4Graph/train/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/train/utils.py -------------------------------------------------------------------------------- /LLM4Graph/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/data_utils.py -------------------------------------------------------------------------------- /LLM4Graph/utils/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/graph.py -------------------------------------------------------------------------------- /LLM4Graph/utils/llm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/utils/llm/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/llm/api.py -------------------------------------------------------------------------------- /LLM4Graph/utils/llm/async_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/llm/async_utils.py -------------------------------------------------------------------------------- /LLM4Graph/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/misc.py -------------------------------------------------------------------------------- /LLM4Graph/utils/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/normalization.py -------------------------------------------------------------------------------- /LLM4Graph/utils/prompt.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LLM4Graph/utils/subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/subgraph.py -------------------------------------------------------------------------------- /LLM4Graph/utils/text_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/text_utils.py -------------------------------------------------------------------------------- /LLM4Graph/utils/tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/LLM4Graph/utils/tuner.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/confs/cora/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/benchmark/confs/cora/nagphormer.yaml -------------------------------------------------------------------------------- /benchmark/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/benchmark/main.py -------------------------------------------------------------------------------- /benchmark/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/benchmark/sweep.py -------------------------------------------------------------------------------- /confs/actor/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/actor/nagphormer.yaml -------------------------------------------------------------------------------- /confs/chameleon/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/chameleon/nagphormer.yaml -------------------------------------------------------------------------------- /confs/citeseer/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/citeseer/nagphormer.yaml -------------------------------------------------------------------------------- /confs/cora/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/cora/nagphormer.yaml -------------------------------------------------------------------------------- /confs/pokec/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/pokec/nagphormer.yaml -------------------------------------------------------------------------------- /confs/products/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/products/nagphormer.yaml -------------------------------------------------------------------------------- /confs/pubmed/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/pubmed/nagphormer.yaml -------------------------------------------------------------------------------- /confs/squirrel/nagphormer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/confs/squirrel/nagphormer.yaml -------------------------------------------------------------------------------- /examples/LLMs_as_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/examples/LLMs_as_predictors.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/main.py -------------------------------------------------------------------------------- /sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CurryTang/LLM4Graph/HEAD/sweep.py --------------------------------------------------------------------------------