-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrun_tdpgd.sh
executable file
·80 lines (68 loc) · 1.98 KB
/
run_tdpgd.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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
model_name=DySAT
dataset=reddit_hyperlinks #auton-sys #opsahl-ucsocial #radoslaw #auton-sys
task=link_prediction #link_prediction
saved_model=models/${model_name}/${dataset}
num_graphs=20 #13 #14
context=19 #12 #13
target_ts=19 #12 #13
ntargets=1
nsamples=50 #100
budget=0
method=pgd
constraint=noise
num_steps=50
lambda1=0
khop=2
nprocs=1
lr_init=10
online=false
if $online; then
file=src/main_online.py
elif [ "$constraint" == noise_feat ]; then
file=src/main_feat.py
else
file=src/main.py
fi
declare -a epsilons_arr=(0.02 0.06 0.1 0.3 0.5 0.7 0.9)
epsilon1=100
resparent_dir="results_${model_name,,}"
if [ ! -d "${resparent_dir}/${dataset}/multi_targets/${method}/${constraint}" ]; then
mkdir -p ${resparent_dir}/${dataset}/multi_targets/${method}/${constraint}
fi
seed=123 #123 483, 665, 811
for epsilon in "${epsilons_arr[@]}"
do
python3 run.py \
-file ${file} \
-constraint ${constraint} \
-budget ${budget} \
-epsilon ${epsilon} \
-epsilon1 ${epsilon1} \
-model_name ${model_name} \
-saved_model ${saved_model} \
-dataset ${dataset} \
-task ${task} \
-num_graphs ${num_graphs} \
-ntargets ${ntargets} \
-khop ${khop} \
-num_steps ${num_steps} \
-lambda1 ${lambda1} \
-context ${context} \
-target_ts ${target_ts} \
-method ${method} \
-featureless \
-num_samples $nsamples \
-device cuda:5 \
-pll_devices cuda:0 cuda:2 cuda:3 cuda:5 \
-neg_sampling \
-seed ${seed} \
-lr_init ${lr_init} \
-nprocs $nprocs \
-use_optim \
-sampling None \
-inits ones > ${resparent_dir}/${dataset}/multi_targets/${method}/${constraint}/results_td_tg${ntargets}_n${num_graphs}_c${context}t${target_ts}_e${epsilon}_eb${epsilon1}_l${lambda1}_seed${seed}.txt
# -dyn_feats \
# b${budget}_l${lambda1}.txt
# -seq_tg_attk \
# -seq_order deg_desc \
done