-
Notifications
You must be signed in to change notification settings - Fork 16
/
runme.sh
117 lines (101 loc) · 4.69 KB
/
runme.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#!/usr/bin/env bash
source activate py37_tf115
export PYTHONPATH=./src # the path of the src folder
DATA_HOME=$1 # the path of the data folder
# train -> the TFRECORD files for training
# valid -> the TFRECORD files for validation
# test -> the TFRECORD files for testing
# mark -> the pickly file which dump scipy.sparse.csr_matrix object
# num_items -> need to specify the number of items
# seqslen -> the maximum length of the sequence
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--mark="${DATA_HOME}/mark.pkl" \
--num_units=512 --hidden_dropout_rate=0.1 --attention_probs_dropout_rate=0.1 \
--learning_rate=5e-4 --batch_size=512 --l2_reg=1e-4 --ct_reg=1e-7 \
--num_items=17771 --model=EasyDGL \
--num_blocks=1 --num_heads=8 --mask_seen --time_scale=86400
#
# Time-independent Baselines,
# GRU4REC, SASREC, GREC, S2PNM, BERT4REC
#
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.2 --attention_probs_dropout_rate=0.2 \
--learning_rate=5e-5 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=BERT4REC \
--num_blocks=3 --num_heads=8 --mask_seen
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0. --attention_probs_dropout_rate=0. \
--learning_rate=5e-4 --batch_size=512 --l2_reg=0. \
--num_items=17771 --model=SASREC \
--num_blocks=2 --num_heads=8 --mask_seen
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.2 --attention_probs_dropout_rate=0.2 \
--learning_rate=1e-4 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=S2PNM \
--num_blocks=1 --num_heads=1 --mask_seen
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.1 \
--learning_rate=5e-4 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=GRU4REC \
--num_blocks=1 --mask_seen
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.1 \
--learning_rate=5e-4 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=GREC \
--dilations=1,4,1,4 --mask_seen
#
# Time-conditioned Baselines,
# TGAT, TiSASREC, TimelyREC, CTSMA
#
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.1 --attention_probs_dropout_rate=0.1 \
--learning_rate=5e-5 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=TGAT \
--num_blocks=3 --num_heads=1 --mask_seen --time_scale=86400
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.1 --attention_probs_dropout_rate=0.1 \
--learning_rate=5e-4 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=TiSASREC --timelen=256\
--num_blocks=2 --num_heads=8 --mask_seen --time_scale=86400
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--num_units=512 --hidden_dropout_rate=0.1 --attention_probs_dropout_rate=0.1 \
--learning_rate=1e-3 --batch_size=512 --l2_reg=1e-4 \
--num_items=17771 --model=TimelyREC \
--num_blocks=2 --num_heads=4 --mask_seen --time_scale=86400
CUDA_VISIBLE_DEVICES=0 python src/main.py \
--train="${DATA_HOME}/train???.tfrec" \
--valid="${DATA_HOME}/validation.tfrec" \
--test="${DATA_HOME}/test.tfrec" \
--mark="${DATA_HOME}/mark.pkl" \
--num_units=512 --hidden_dropout_rate=0.1 --attention_probs_dropout_rate=0.2 \
--learning_rate=5e-4 --batch_size=512 --l2_reg=1e-4 --ct_reg=1e-7 \
--num_items=17771 --model=CTSMA \
--num_blocks=2 --num_heads=4 --mask_seen --time_scale=86400
conda deactivate