-
Notifications
You must be signed in to change notification settings - Fork 32
/
Copy pathconfig.json
42 lines (42 loc) · 804 Bytes
/
config.json
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
{
"train_config": {
"batch_size": 256,
"num_epochs": 100,
"train_ratio": 0.9,
"learning_rate": 0.001,
"optimizer": "adam",
"seq_len": 100
},
"dkt": {
"emb_size": 100,
"hidden_size": 100
},
"dkt+": {
"emb_size": 100,
"hidden_size": 100,
"lambda_r": 0.01,
"lambda_w1": 0.003,
"lambda_w2": 3.0
},
"dkvmn": {
"dim_s": 50,
"size_m": 20
},
"sakt": {
"n": 100,
"d": 100,
"num_attn_heads": 5,
"dropout": 0.2
},
"saint": {
"n": 100,
"d": 30,
"num_attn_heads": 5,
"dropout": 0.6
},
"gkt": {
"hidden_size": 30,
"num_attn_heads": 2,
"method": "PAM"
}
}