-
Notifications
You must be signed in to change notification settings - Fork 6
/
run_graph_tokenizer.sh
29 lines (24 loc) · 1.01 KB
/
run_graph_tokenizer.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
cd /path/to/HiGPT
# offline tokenizing for stage1 matching instruction
data_type=imdb
graph_root=./hi_datasets/matching_instruction
dsname_list=(instruct_ds_matching_movie instruct_ds_node_matching)
pretrained_hgt=./MetaHGT_imdb_dblp_epoch5
for dsname in "${dsname_list[@]}"
do
python run_offline_hgt_tokenizer.py --dsname ${dsname} \
--data_type ${data_type} \
--pretrained_gnn_path ${pretrained_hgt} \
--graph_root ${graph_root}
done
data_type=dblp
graph_root=./hi_datasets/matching_instruction
dsname_list=(instruct_ds_matching_author instruct_ds_matching_paper instruct_ds_node_matching)
pretrained_hgt=./MetaHGT_imdb_dblp_epoch5
for dsname in "${dsname_list[@]}"
do
python run_offline_hgt_tokenizer.py --dsname ${dsname} \
--data_type ${data_type} \
--pretrained_gnn_path ${pretrained_hgt} \
--graph_root ${graph_root}
done