-
Notifications
You must be signed in to change notification settings - Fork 0
/
params.inr_only.yaml
107 lines (102 loc) · 1.74 KB
/
params.inr_only.yaml
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
101
102
103
104
105
106
107
preprocess:
static_state_columns:
- SEX
- RACE2
- CONTINENT
- SMOKE
- BMED_ASPIRIN
- BMED_AMIOD
- DIABETES
- HX_CHF
- HYPERTENSION
- HX_MI
- BMED_THIENO
- AGE_DEIDENTIFIED
- WEIGHT
dose_outlier_threshold: 140
event_range: 90
max_time_elapsed: 90
data:
min_train_trajectory_length: 5
events_to_keep:
DEATH: Death
STROKE: Ischemic Stroke
MAJOR_BLEED: Major Bleeding
MINOR_BLEED: Minor Bleeding
HEM_STROKE: Hemorrhagic Stroke
HOSP: Hospitalization
SYS_EMB: Systemic Embolism
events_to_split:
- DEATH
- STROKE
- MAJOR_BLEED
- HEM_STROKE
- HOSP
- SYS_EMB
events_to_evaluate:
- STROKE
- HEM_STROKE
- MAJOR_BLEED
state_columns:
- INR_VALUE
replay_buffer:
init:
time_varying: across
include_dose_time_varying: false
include_duration_time_varying: false
discount_factor: 0.99
inr_reward: 1
event_reward: 0
behavior_cloner:
tune_seed: 42
min_training_epochs: 100
max_training_epochs: 500
target_metric: val/auroc
target_mode: max
hyperparams:
init_seed:
- 0
- 1
- 2
likelihood:
- discrete
learning_rate:
- 1.e-4
- 1.e-3
batch_size:
- 16
num_layers:
- 2
hidden_dim:
- 64
weight_option_frequency:
- false
dsbcq:
tune_seed: 43
min_training_epochs: 50
max_training_epochs: 2500
plot_every: 100
all_metrics_every: 100
target_metric: val/wis/policy_value
target_mode: max
hyperparams:
init_seed:
- 1
- 2
bcq_threshold:
- 0.3
- 0.4
batch_size:
- 32
num_layers:
- 2
hidden_dim:
- 64
learning_rate:
- 1.e-5
tau:
- 5.e-3
dsbcq_eval:
min_training_epochs: 500
model_sel_metric: val/wis/policy_value
model_sel_mode: max