[d6904d]: / configs / get_model_output.yaml

Download this file

14 lines (11 with data), 336 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
name: get_output
model_type: ehr
model: adacare
task: twostage
# dataset: tongji
# outcome_model_name: tj_outcome_agent_ep100_kf10_bs64_hid128
# los_model_name: tj_los_mlp_ep100_kf10_bs64_hid128
dataset: hm
outcome_model_name: hm_outcome_adacare_ep100_kf10_bs512_hid64
los_model_name: hm_los_mlp_ep100_kf10_bs512_hid128
train_fold: 1