M3E2/config/config2a.yaml at main · raquelaoki/M3E2 · 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
#Parameters
parameters:
    data: ['copula'] #gwas, copula
    n_sample: 2500
    n_covariates: 10
    n_treatments: 4
    model: 'm3e2'
    id: 'config2a'
    suffle: True
    best_validation_test: True
    batch_size: 100
    max_epochs: 100
    print: 50
    decay: 20
    alpha: 0.01 #decay downgrad
    gamma: 0.8 #Adam decay
    lr: 0.001 #learning rate
    wd: 0.05 #weight_decay
    dropoutp: 0.5
    pos_weight_y: 0.7
    pos_weights: [ 0.7, 1, 1,1 ]
    num_exp: 2
    units_exp: 4
    hidden1: 6
    hidden2: 3
    is_binary_treatment: False
    is_binary_target: True
    baselines: True
    baselines_list: ['DA','CEVAE','Dragonnet']
    loss_target: 6
    loss_da: 2
    loss_treat: 0.5
    loss_reg: 1
    da_k: [ 5 ]
    CEVAE_z_dim: 5
    Dragonnet_u1: 200
    Dragonnet_u2: 100
    Dragonnet_u3: 1