-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_dblp.sh
30 lines (22 loc) · 940 Bytes
/
run_dblp.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
#!/bin/bash
cecho(){
RED="\033[0;31m"
YELLOW="\033[1;33m"
NC="\033[0m" # No Color
printf "${!1}${2} ${NC}\n"
}
gpu=${1:-0}
data="dblp"
we=300
ade=20
v="ratio_0.001 apa apcpa aptpa"
python make_graph.py --dataset ${data}
for seed in 3 6 9 12 15
do
cecho "RED" "Start Dataset=${data} Seed=${seed} Num=20 View=$v"
python train.py --dataset ${data} --seed ${seed} --contrastive --view ${v} --lam 0.2 --lr 0.0001 --gpu ${gpu} --num 20 --adjust_epoch ${ade} --warm_epoch ${we}
cecho "RED" "Start Dataset=${data} Seed=${seed} Num=40 View=$v"
python train.py --dataset ${data} --seed ${seed} --contrastive --view ${v} --lam 1 --lr 0.0001 --gpu ${gpu} --num 40 --adjust_epoch ${ade} --warm_epoch ${we}
cecho "RED" "Start Dataset=${data} Seed=${seed} Num=60 View=$v"
python train.py --dataset ${data} --seed ${seed} --contrastive --view ${v} --lam 0.2 --lr 0.0001 --gpu ${gpu} --num 60 --adjust_epoch ${ade} --warm_epoch ${we}
done