File size: 1,353 Bytes
2de08f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
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
|