Downloads: 1
Diff of
/configs/get_model_output.yaml
[000000]
..
[d6904d]
Switch to unified view
| a | b/configs/get_model_output.yaml | ||
|---|---|---|---|
| 1 | name: get_output |
||
| 2 | model_type: ehr |
||
| 3 | model: adacare |
||
| 4 | task: twostage |
||
| 5 | |||
| 6 | # dataset: tongji |
||
| 7 | # outcome_model_name: tj_outcome_agent_ep100_kf10_bs64_hid128 |
||
| 8 | # los_model_name: tj_los_mlp_ep100_kf10_bs64_hid128 |
||
| 9 | |||
| 10 | dataset: hm |
||
| 11 | outcome_model_name: hm_outcome_adacare_ep100_kf10_bs512_hid64 |
||
| 12 | los_model_name: hm_los_mlp_ep100_kf10_bs512_hid128 |
||
| 13 | |||
| 14 | train_fold: 1 |