-
Notifications
You must be signed in to change notification settings - Fork 19
/
run_main.sh
executable file
·74 lines (55 loc) · 3.23 KB
/
run_main.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
#!/bin/bash
run_commands(){
eval template="$1"
for train_frac in 0.5 0.4 0.3 0.2 0.1; do
for ((i=1; i<=10; i++)); do
run_param="${i}o10"
eval "$1 --run $run_param --train_frac $train_frac"
done
done
}
cd src/
# Strats mimic_iii
python main.py --pretrain 1 --dataset mimic_iii --model_type strats --hid_dim 64 --num_layers 2 --num_heads 16 --dropout 0.2 --attention_dropout 0.2 --lr 5e-4 --max_epochs 30
template="python main.py --dataset mimic_iii --model_type strats --hid_dim 64 --num_layers 2 --num_heads 16 --dropout 0.2 --attention_dropout 0.2 --lr 5e-5 --load_ckpt_path ../outputs/mimic_iii/pretrain/checkpoint_best.bin"
run_commands "\${template}"
# Strats physionet
python main.py --pretrain 1 --dataset physionet_2012 --model_type strats --hid_dim 64 --num_layers 2 --num_heads 16 --dropout 0.2 --attention_dropout 0.2 --lr 5e-4 --max_epochs 100
template="python main.py --dataset physionet_2012 --model_type strats --hid_dim 64 --num_layers 2 --num_heads 16 --dropout 0.2 --attention_dropout 0.2 --lr 5e-5 --load_ckpt_path ../outputs/physionet_2012/pretrain/checkpoint_best.bin"
run_commands "\${template}"
# Strats (ss-) mimic_iii
template="python main.py --dataset mimic_iii --model_type strats --hid_dim 64 --num_layers 2 --num_heads 16 --dropout 0.2 --attention_dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# Strats (ss-) physionet
template="python main.py --dataset physionet_2012 --model_type strats --hid_dim 64 --num_layers 2 --num_heads 16 --dropout 0.2 --attention_dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# TCN physionet
template="python main.py --dataset physionet_2012 --model_type tcn --num_layers 6 --hid_dim 64 --kernel_size 4 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# GRUD physionet
template="python main.py --dataset physionet_2012 --model_type grud --hid_dim 64 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# GRU physionet
template="python main.py --dataset physionet_2012 --model_type gru --hid_dim 64 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# SAND physionet
template="python main.py --dataset physionet_2012 --model_type sand --num_layers 4 --r 24 --M 12 --hid_dim 64 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# Interpnet physionet
template="python main.py --dataset physionet_2012 --model_type interpnet --hid_dim 64 --ref_points 192 --hours_look_ahead 48 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# TCN mimic_iii
template="python main.py --dataset mimic_iii --model_type tcn --num_layers 4 --hid_dim 128 --kernel_size 4 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# GRU mimic_iii
template="python main.py --dataset mimic_iii --model_type gru --hid_dim 64 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# SAND mimic_iii
template="python main.py --dataset mimic_iii --model_type sand --num_layers 4 --r 24 --M 12 --hid_dim 64 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# Interpnet mimic_iii
template="python main.py --dataset mimic_iii --model_type interpnet --hid_dim 64 --ref_points 96 --hours_look_ahead 24 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"
# GRUD mimic_iii
template="python main.py --dataset mimic_iii --model_type grud --hid_dim 64 --dropout 0.2 --lr 5e-4"
run_commands "\${template}"