TEGA/example.sh at main · HKUST-KnowComp/TEGA · GitHub

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
30
31
32
    python train.py \
    -dn FB15k-237 \
    -m transformer \
    --train_query_dir ../sample_query_data/fb237-23-train \
    --valid_query_dir ../sample_query_data/fb237-55-valid \
    --test_query_dir ../sample_query_data/fb237-55-test \
    --checkpoint_path ../checkpoint/logs \
    -b 5 \
    --log_steps 2000 \
    -lr 0.0001 \
    --rpe


    python train.py \
    -dn FB15k-237 \
    -m transformertega \
    --train_query_dir ../sample_query_data/fb237-23-train \
    --valid_query_dir ../sample_query_data/fb237-55-valid \
    --test_query_dir ../sample_query_data/fb237-55-test \
    --checkpoint_path ../checkpoint/logs \
    -b 5 \
    --log_steps 2000 \
    -lr 0.0001 \
    --rpe \
    --num_categories 6 \
    --pooling sum \
    --self_dist

python sample_query_multi.py --sample_formula_scope FULL55 --mode test --output_folder ../your_folder_name --num_queries 100 --num_processes 20
python sample_query_multi.py --sample_formula_scope FULL55 --mode valid --output_folder ../your_folder_name --num_queries 100 --num_processes 20
python sample_query_multi.py --sample_formula_scope SEEN23 --mode train --output_folder ../your_folder_name --num_queries 200 --num_processes 20