File size: 1,128 Bytes
2de08f8 |
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 |
data:
path_table: ./data/sick_train.csv
numerical_columns:
- age
- TSH
- T3
- TT4
- T4U
- FTI
categorical_columns:
- Class
- sex
- on_thyroxine
- query_on_thyroxine
- on_antithyroid_medication
- sick
- pregnant
- thyroid_surgery
- I131_treatment
- query_hypothyroid
- query_hyperthyroid
- lithium
- goitre
- tumor
- hypopituitary
- psych
- TSH_measured
- T3_measured
- TT4_measured
- T4U_measured
- FTI_measured
- referral_source
columns_to_drop:
- TBG
- TBG_measured
dropna: true
fillna: false
target_column: Class
split_feature_target: true
task: classification
model:
dim: 256
n_res_blocks: 3
diffusion:
schedule: quad
n_timesteps: 1000
target: two_way
trainer:
train_num_steps: 500000
log_every: 100
save_every: 10000
save_num_samples: 64
max_grad_norm: null
gradient_accumulate_every: 1
ema_decay: 0.995
ema_update_every: 10
lr: 0.0001
opt_type: adam
opt_params: null
batch_size: 256
dataloader_workers: 16
classifier_free_guidance: true
zero_token_probability: 0.1
fine_tune_from: null
comment: sick_CFG
|