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