Towards Efficient LLMs-aware Heterogeneous Graph Learning

Official codebase for paper Towards Efficient LLMs-aware Heterogeneous Graph Learning.

1. Data Preparation

Place the downloaded Llama2-13b model in the ../Data/Llama213b folder.

2. Training ELLA

cd Data_Preprocess
sh run.sh
cd ../
cd HetPrompt/pretrain_0/
sh run.sh

Please note that "pretrain_specific_name_lst" in "HetPrompt/finetune_1/scripts-/run_.sh" needs to be configured as the filename output from the pre-training stage.

cd ../
cd finetune_1/
sh run.sh