| 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 | |