| data: | |
| path_table: ./data/heloc_train.csv | |
| numerical_columns: | |
| - ExternalRiskEstimate | |
| - MSinceOldestTradeOpen | |
| - MSinceMostRecentTradeOpen | |
| - AverageMInFile | |
| - NumSatisfactoryTrades | |
| - NumTrades60Ever2DerogPubRec | |
| - NumTrades90Ever2DerogPubRec | |
| - PercentTradesNeverDelq | |
| - MSinceMostRecentDelq | |
| - MaxDelq2PublicRecLast12M | |
| - MaxDelqEver | |
| - NumTotalTrades | |
| - NumTradesOpeninLast12M | |
| - PercentInstallTrades | |
| - MSinceMostRecentInqexcl7days | |
| - NumInqLast6M | |
| - NumInqLast6Mexcl7days | |
| - NetFractionRevolvingBurden | |
| - NetFractionInstallBurden | |
| - NumRevolvingTradesWBalance | |
| - NumInstallTradesWBalance | |
| - NumBank2NatlTradesWHighUtilization | |
| - PercentTradesWBalance | |
| categorical_columns: | |
| - RiskPerformance | |
| columns_to_drop: null | |
| dropna: true | |
| fillna: false | |
| target_column: RiskPerformance | |
| 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: heloc_CFG | |