ZNet/bayesian_bounds_znet.json at main · AlaaLab/ZNet · 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
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
{
    "znet_params": {
        "lr": [1e-5, 1e-2],
        "kl_loss_coeff": [0, 1e-1],
        "feature_corr_loss_coeff": [0, 1e-1],
        "c_pearson_loss_alpha": [0, 1],
        "z_pearson_loss_alpha": [0.5, 1],
        "z_t_loss_alpha": [0, 1],
        "c_mse_loss_alpha": [0.5, 1],
        "pearson_matrix_alpha": [0, 1],
        "t_hat_alpha": [0, 1],
        "use_pcgrad": [0, 0],
        "is_linear": [0, 1],
        "use_sm": [0, 1],
        "sm_temp": [0, 3],
        "use_mi_corr_loss": [0, 0],
        "use_mi_matrix_loss": [0, 0]
    },
    "znet_train_params": {
        "batch_size": [32, 256],
        "num_epochs": [25, 75]
    },
    "df_iv_params": {
        "batch_size": [64, 128],
        "dropout": [0, 0.5],
        "hidden_size_phi": [20, 40],
        "hidden_size_psi": [20, 40],
        "hidden_size_xi": [20, 40],
        "lambda1": [1e-5, 1e-2],
        "lambda2": [1e-5, 1e-2],
        "lr1": [1e-5, 1e-2],
        "lr2": [1e-5, 1e-2],
        "epochs": [100, 200],
        "use_early_stopping": [0, 1],
        "patience": [5, 20]
    },
    "deep_iv_params": {
        "batch_size": [64, 128],
        "batch_size2": [64, 128],
        "dropout": [0, 0.5],
        "dropout2": [0, 0.5],
        "hidden_size": [20, 40],
        "hidden_size2": [20, 40],
        "lr": [1e-5, 1e-2],
        "lr2": [1e-5, 1e-2],
        "epochs1": [100, 200],
        "epochs2": [100, 200],
        "use_early_stopping": [0, 1],
        "patience": [5, 20]
    },
    "dim_options": {
        "c_dim": [5, 15],
        "z_dim": [1, 10]
    },
    "autoiv_params": {
        "emb_dim": [1, 10],
        "rep_dim": [1, 10],
        "epochs": [100, 300],
        "sigma": [0.01, 0.1],
        "lrate": [1e-5, 1e-2],
        "coef_cx2y": [0, 1],
        "coef_zc2x": [0, 1],
        "coef_lld_zx": [0, 1],
        "coef_lld_zy": [0, 1],
        "coef_lld_cx": [0, 1],
        "coef_lld_cy": [0, 1],
        "coef_lld_zc": [0, 1],
        "coeff_reg": [0, 1],
        "dropout": [0, 0.5]
    },
    "giv_params": {
        "num_clusters": [2, 10],
        "batch_size": [100, 400],
        "epoch": [100, 300],
        "beta1": [0.7, 0.9],
        "beta2": [0.9, 0.999],
        "lr": [1e-4, 1e-3]
    },
    "viv_params": {
        "epochs": [100, 200],
        "bs": [64, 128],
        "d": [10, 50],
        "lrate": [1e-5, 1e-2],
        "lrate_min": [1e-6, 1e-4],
        "loss_y": [0.1, 0.5],
        "loss_t": [0.1, 0.5],
        "loss_x": [0.1, 0.5],
        "kl_loss": [0.1, 0.5],
        "ad_loss": [0.5, 1]
    },
     "tarnet_params": {
        "weight_decay" : [0, 0.01],
        "lr" : [1e-5, 1e-2]
    },

    "tarnet_train_params": {
        "batch_size" : [64, 128],
        "num_epochs" : [100, 200]
    }
}