forked from Namkyeong/AFGRL
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_similarity_search.sh
17 lines (15 loc) · 2.06 KB
/
run_similarity_search.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# python main.py --dataset wikics --embedder AFGRL --task node --layers [1024] --pred_hid 2048 --lr 0.001 --topk 8 --device 0 --mad 0.9
# python main.py --dataset computers --embedder AFGRL --task node --layers [512] --pred_hid 1024 --lr 0.001 --topk 4 --device 0 --mad 0.9
# python main.py --dataset photo --embedder AFGRL --task node --layers [512] --pred_hid 1024 --lr 0.001 --topk 4 --device 1 --mad 0.9
# python main.py --dataset cs --embedder AFGRL --task node --layers [1024] --pred_hid 2048 --lr 0.001 --topk 4 --device 1 --mad 0.9 --es 1000
# python main.py --dataset physics --embedder AFGRL --task node --layers [256] --pred_hid 512 --lr 0.01 --topk 8 --device 0 --mad 0.9 --es 1000
# python main.py --dataset wikics --embedder AFGRL --task clustering --layers [1024] --pred_hid 2048 --lr 0.001 --topk 8 --device 0 --mad 0.9
# python main.py --dataset computers --embedder AFGRL --task clustering --layers [512] --pred_hid 1024 --lr 0.001 --topk 4 --device 0 --mad 0.9
# python main.py --dataset photo --embedder AFGRL --task clustering --layers [512] --pred_hid 1024 --lr 0.001 --topk 4 --device 0 --mad 0.9
# python main.py --dataset cs --embedder AFGRL --task clustering --layers [1024] --pred_hid 2048 --lr 0.001 --topk 4 --device 0 --mad 0.9
# python main.py --dataset physics --embedder AFGRL --task clustering --layers [256] --pred_hid 512 --lr 0.01 --topk 8 --device 0 --mad 0.9
python main.py --dataset wikics --embedder AFGRL --task similarity --layers [1024] --pred_hid 2048 --lr 0.001 --topk 8 --device 0 --mad 0.9
python main.py --dataset computers --embedder AFGRL --task similarity --layers [512] --pred_hid 1024 --lr 0.001 --topk 4 --device 0 --mad 0.9
python main.py --dataset photo --embedder AFGRL --task similarity --layers [512] --pred_hid 1024 --lr 0.001 --topk 4 --device 1 --mad 0.9
python main.py --dataset cs --embedder AFGRL --task similarity --layers [1024] --pred_hid 2048 --lr 0.001 --topk 4 --device 1 --mad 0.9 --es 1000
python main.py --dataset physics --embedder AFGRL --task similarity --layers [256] --pred_hid 512 --lr 0.01 --topk 8 --device 0 --mad 0.9 --es 1000