-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
69 lines (65 loc) · 1.37 KB
/
run.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
DATASET=$1
if [[ "$DATASET" == "swat" ]]; then
WINDOW_SIZE=50
HORIZON=1
STRIDE=1
BATCH_SIZE=64
EMBED_DIM=32
TOPK=5
EPOCHS=50
VAL_SPLIT=0.2
EARLY_STOPPING=10
SMOOTHING=1
SMOOTHING_METHOD="exp"
THRESHOLDING="best"
TRANSFORM="median"
TARGET_TRANSFORM="median"
NORMALIZE="True"
elif [[ "$DATASET" == "wadi" ]]; then
WINDOW_SIZE=50
STRIDE=1
HORIZON=1
BATCH_SIZE=64
EMBED_DIM=32
TOPK=8
EPOCHS=50
VAL_SPLIT=0.1
EARLY_STOPPING=10
SMOOTHING=1
SMOOTHING_METHOD="exp"
THRESHOLDING="best"
TRANSFORM="median"
TARGET_TRANSFORM="median"
NORMALIZE="True"
elif [[ "$DATASET" == "demo" ]]; then
WINDOW_SIZE=25
STRIDE=1
HORIZON=1
BATCH_SIZE=32
EMBED_DIM=16
TOPK=3
EPOCHS=50
VAL_SPLIT=0
EARLY_STOPPING=20
SMOOTHING=1
SMOOTHING_METHOD="mean"
THRESHOLDING="best"
TRANSFORM="none"
TARGET_TRANSFORM="none"
NORMALIZE="False"
fi
python main.py \
-dataset $DATASET \
-window_size $WINDOW_SIZE \
-horizon $HORIZON \
-stride $STRIDE \
-val_split $VAL_SPLIT \
-batch_size $BATCH_SIZE \
-embed_dim $EMBED_DIM \
-topk $TOPK \
-epochs $EPOCHS \
-early_stopping $EARLY_STOPPING \
-smoothing $SMOOTHING \
-smoothing_method $SMOOTHING_METHOD \
-thresholding $THRESHOLDING \
-normalize $NORMALIZE \