| data: | |
| path_table: ./data/travel_train.csv | |
| numerical_columns: | |
| - Age | |
| - ServicesOpted | |
| categorical_columns: | |
| - Target | |
| - FrequentFlyer | |
| - AnnualIncomeClass | |
| - AccountSyncedToSocialMedia | |
| - BookedHotelOrNot | |
| columns_to_drop: null | |
| dropna: true | |
| fillna: false | |
| target_column: Target | |
| 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: 100 | |
| 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: travel_CFG | |