HiSACKT/config.json at main · duaabaig/HiSACKT · GitHub

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
{
    "train_config": {
        "batch_size": 256,
        "num_epochs": 100,
        "train_ratio": 0.7,
        "val_ratio": 0.15,
        "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
    },

    "gkt": {
        "hidden_size": 30,
        "num_attn_heads": 2,
        "method": "MHA"
    },

    "hisackt": {
        "n": 100,
        "d": 100,
        "num_attn_heads": 2,
        "ha_num_heads_per_level": [2, 2, 2, 2],
        "local_window": 25,
        "dropout": 0.2
    }
}