Upload models, transformations and training configurations

adult/config.yaml

@@ -0,0 +1,49 @@
+data:
+  path_table: ./data/adult_train.csv
+  numerical_columns:
+  - age
+  - fnlwgt
+  - education-num
+  - capital-gain
+  - capital-loss
+  - hours-per-week
+  categorical_columns:
+  - workclass
+  - education
+  - marital-status
+  - occupation
+  - relationship
+  - race
+  - sex
+  - native-country
+  columns_to_drop: null
+  dropna: true
+  fillna: false
+  target_column: label
+  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: adult_CFG

adult/model-final.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51b042e9ee79e20dc2586506495d0ee938d6e6317d630e8a9b6beb6392ca01cb
+size 22952666

adult/transformation.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28184b8daa946968fddfd0f06bcb978fda0e90ad6bd7a6d7766db1235123041e
+size 3083

diabetes/config.yaml

@@ -0,0 +1,84 @@
+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

diabetes/model-final.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea60904d1a60d3eb7692841fc91b3714a7f7f65486bad0c822c9710fed4c7464
+size 28343578

diabetes/transformation.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8d8b53f148f4488ae70bb332650c8ddafe1c56bb917c53ff8a7983f23ced502e
+size 6606

heloc/config.yaml

@@ -0,0 +1,59 @@
+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

heloc/model-final.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f11f653a2ee423cccb8f5c17e7dfb673babb74814aa9171709ae4e053da6f57b
+size 41991194

heloc/transformation.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1e33cfa1789184a8baffb1b182440288069f7e562794762cf773caf05ed6de6f
+size 3177

housing/config.yaml

@@ -0,0 +1,45 @@
+data:
+  path_table: ./data/housing_train.csv
+  numerical_columns:
+  - longitude
+  - latitude
+  - housing_median_age
+  - total_rooms
+  - total_bedrooms
+  - population
+  - households
+  - median_income
+  - median_house_value
+  categorical_columns:
+  - ocean_proximity
+  columns_to_drop: null
+  dropna: true
+  fillna: false
+  target_column: median_house_value
+  split_feature_target: true
+  task: regression
+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: housing_CFG

housing/model-final.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a3c41a9eb22c34f033dc7fe733798ff3d805ea9b4b2c18cd95da50491d080c12
+size 24387610

housing/transformation.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50bfd88004e7c329c4cfc37bba9cf8d1f9ec25a8eaa87c6ca1ce145a22b81725
+size 1776

sick/config.yaml

@@ -0,0 +1,65 @@
+data:
+  path_table: ./data/sick_train.csv
+  numerical_columns:
+  - age
+  - TSH
+  - T3
+  - TT4
+  - T4U
+  - FTI
+  categorical_columns:
+  - Class
+  - sex
+  - on_thyroxine
+  - query_on_thyroxine
+  - on_antithyroid_medication
+  - sick
+  - pregnant
+  - thyroid_surgery
+  - I131_treatment
+  - query_hypothyroid
+  - query_hyperthyroid
+  - lithium
+  - goitre
+  - tumor
+  - hypopituitary
+  - psych
+  - TSH_measured
+  - T3_measured
+  - TT4_measured
+  - T4U_measured
+  - FTI_measured
+  - referral_source
+  columns_to_drop:
+  - TBG
+  - TBG_measured
+  dropna: true
+  fillna: false
+  target_column: Class
+  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: sick_CFG

sick/model-final.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:defbcc16a6e3d833e7272db8baee5857f7aee20e58fda9d6433fd15b40058fc6
+size 22805210

sick/transformation.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f997f4b14be9dcee192af00a214b1d3c54659992c04ae6f9d7f0a53416d9c92
+size 2333

travel/config.yaml

@@ -0,0 +1,42 @@
+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

travel/model-final.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ce1267cbd1ec0c18a51fbfeb1925b484e4c058de26de4e0deca4282b1d27d19
+size 17418202

travel/transformation.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c323800874f43e9615a656cd62797974fa8432c132c2b1a3b094806e17660474
+size 1122