| data: | |
| path_table: ./data/diabetes_train.csv | |
| numerical_columns: | |
| - num_lab_procedures | |
| - num_procedures | |
| - num_medications | |
| - number_outpatient | |
| - number_emergency | |
| - number_inpatient | |
| - number_diagnoses | |
| - time_in_hospital | |
| categorical_columns: | |
| - race | |
| - gender | |
| - age | |
| - weight | |
| - admission_type_id | |
| - discharge_disposition_id | |
| - admission_source_id | |
| - payer_code | |
| - medical_specialty | |
| - diag_1 | |
| - diag_2 | |
| - diag_3 | |
| - max_glu_serum | |
| - A1Cresult | |
| - metformin | |
| - repaglinide | |
| - nateglinide | |
| - chlorpropamide | |
| - glimepiride | |
| - acetohexamide | |
| - glipizide | |
| - glyburide | |
| - tolbutamide | |
| - pioglitazone | |
| - rosiglitazone | |
| - acarbose | |
| - miglitol | |
| - troglitazone | |
| - tolazamide | |
| - examide | |
| - citoglipton | |
| - insulin | |
| - glyburide-metformin | |
| - glipizide-metformin | |
| - glimepiride-pioglitazone | |
| - metformin-rosiglitazone | |
| - metformin-pioglitazone | |
| - change | |
| - diabetesMed | |
| columns_to_drop: | |
| - encounter_id | |
| - patient_nbr | |
| dropna: true | |
| fillna: false | |
| target_column: readmitted | |
| 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: diabetes_CFG | |