├── CTABGANPlus ├── ctabgan.py ├── pipeline │ └── data_preparation.py └── synthesizer │ ├── ctabgan_synthesizer.py │ └── transformer.py ├── Python-Package └── base-ForestDiffusion │ ├── ForestDiffusion │ ├── __init__.py │ ├── diffusion_with_trees_class.py │ └── utils │ │ ├── __init__.py │ │ ├── diffusion.py │ │ └── utils_diffusion.py │ ├── LICENSE.txt │ └── setup.py ├── R-Package ├── Documentation.pdf ├── ForestDiffusion │ ├── DESCRIPTION │ ├── LICENSE │ ├── NAMESPACE │ ├── NEWS │ ├── R │ │ └── ForestDiffusion.R │ ├── man │ │ ├── ForestDiffusion.Rd │ │ ├── ForestDiffusion.generate.Rd │ │ ├── ForestDiffusion.impute.Rd │ │ └── with_datasets.Rd │ └── vignettes │ │ └── ForestDiffusion.Rmd ├── ForestDiffusion_1.0.0.tar.gz ├── Vignette.html ├── manual_installation.R └── test.R ├── README.md ├── Results ├── Results_tabular - ablation_iris.csv ├── Results_tabular - generation.csv ├── Results_tabular - generation_miss.csv ├── Results_tabular - imputation.csv ├── generation_script.R ├── generation_script_nmiss50.R └── imputation_script.R ├── STaSy ├── datasets.py ├── datasets_tabular.py ├── likelihood.py ├── losses.py ├── models │ ├── __init__.py │ ├── ema.py │ ├── layers.py │ ├── layerspp.py │ ├── ncsnpp_tabular.py │ ├── tabular_utils.py │ └── utils.py ├── sampling.py ├── sde_lib.py ├── stasy.py └── utils.py ├── TabDDPM ├── CONFIG_DESCRIPTION.md ├── LICENSE.md ├── README.md ├── agg_results.ipynb ├── config │ ├── config.toml │ └── config_test.toml ├── lib │ ├── __init__.py │ ├── data.py │ ├── dataset_prep.py │ ├── deep.py │ ├── env.py │ ├── metrics.py │ └── util.py ├── requirements.txt ├── scripts │ ├── __init__.py │ ├── pipeline.py │ ├── sample.py │ ├── train.py │ └── utils_train.py └── tab_ddpm │ ├── __init__.py │ ├── gaussian_multinomial_diffsuion.py │ ├── modules.py │ └── utils.py ├── data_loaders.py ├── download_data.py ├── fi_shap.png ├── gain.py ├── imputers.py ├── iris_cropped.png ├── metrics.py ├── missforest.py ├── requirements.txt ├── script_generation.py ├── script_imputation.py ├── softimpute.py └── utils.py /CTABGANPlus/ctabgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/CTABGANPlus/ctabgan.py -------------------------------------------------------------------------------- /CTABGANPlus/pipeline/data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/CTABGANPlus/pipeline/data_preparation.py -------------------------------------------------------------------------------- /CTABGANPlus/synthesizer/ctabgan_synthesizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/CTABGANPlus/synthesizer/ctabgan_synthesizer.py -------------------------------------------------------------------------------- /CTABGANPlus/synthesizer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/CTABGANPlus/synthesizer/transformer.py -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/ForestDiffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/ForestDiffusion/__init__.py -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/ForestDiffusion/diffusion_with_trees_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/ForestDiffusion/diffusion_with_trees_class.py -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/ForestDiffusion/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/ForestDiffusion/utils/__init__.py -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/ForestDiffusion/utils/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/ForestDiffusion/utils/diffusion.py -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/ForestDiffusion/utils/utils_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/ForestDiffusion/utils/utils_diffusion.py -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/LICENSE.txt -------------------------------------------------------------------------------- /Python-Package/base-ForestDiffusion/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Python-Package/base-ForestDiffusion/setup.py -------------------------------------------------------------------------------- /R-Package/Documentation.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/Documentation.pdf -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/DESCRIPTION: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/DESCRIPTION -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/LICENSE: -------------------------------------------------------------------------------- 1 | YEAR: 2023 2 | COPYRIGHT HOLDER: Samsung Electronics Co., Ltd. -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/NAMESPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/NAMESPACE -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/NEWS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/NEWS -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/R/ForestDiffusion.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/R/ForestDiffusion.R -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/man/ForestDiffusion.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/man/ForestDiffusion.Rd -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/man/ForestDiffusion.generate.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/man/ForestDiffusion.generate.Rd -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/man/ForestDiffusion.impute.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/man/ForestDiffusion.impute.Rd -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/man/with_datasets.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/man/with_datasets.Rd -------------------------------------------------------------------------------- /R-Package/ForestDiffusion/vignettes/ForestDiffusion.Rmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion/vignettes/ForestDiffusion.Rmd -------------------------------------------------------------------------------- /R-Package/ForestDiffusion_1.0.0.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/ForestDiffusion_1.0.0.tar.gz -------------------------------------------------------------------------------- /R-Package/Vignette.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/Vignette.html -------------------------------------------------------------------------------- /R-Package/manual_installation.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/manual_installation.R -------------------------------------------------------------------------------- /R-Package/test.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/R-Package/test.R -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/README.md -------------------------------------------------------------------------------- /Results/Results_tabular - ablation_iris.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/Results_tabular - ablation_iris.csv -------------------------------------------------------------------------------- /Results/Results_tabular - generation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/Results_tabular - generation.csv -------------------------------------------------------------------------------- /Results/Results_tabular - generation_miss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/Results_tabular - generation_miss.csv -------------------------------------------------------------------------------- /Results/Results_tabular - imputation.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/Results_tabular - imputation.csv -------------------------------------------------------------------------------- /Results/generation_script.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/generation_script.R -------------------------------------------------------------------------------- /Results/generation_script_nmiss50.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/generation_script_nmiss50.R -------------------------------------------------------------------------------- /Results/imputation_script.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/Results/imputation_script.R -------------------------------------------------------------------------------- /STaSy/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/datasets.py -------------------------------------------------------------------------------- /STaSy/datasets_tabular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/datasets_tabular.py -------------------------------------------------------------------------------- /STaSy/likelihood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/likelihood.py -------------------------------------------------------------------------------- /STaSy/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/losses.py -------------------------------------------------------------------------------- /STaSy/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/__init__.py -------------------------------------------------------------------------------- /STaSy/models/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/ema.py -------------------------------------------------------------------------------- /STaSy/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/layers.py -------------------------------------------------------------------------------- /STaSy/models/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/layerspp.py -------------------------------------------------------------------------------- /STaSy/models/ncsnpp_tabular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/ncsnpp_tabular.py -------------------------------------------------------------------------------- /STaSy/models/tabular_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/tabular_utils.py -------------------------------------------------------------------------------- /STaSy/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/models/utils.py -------------------------------------------------------------------------------- /STaSy/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/sampling.py -------------------------------------------------------------------------------- /STaSy/sde_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/sde_lib.py -------------------------------------------------------------------------------- /STaSy/stasy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/stasy.py -------------------------------------------------------------------------------- /STaSy/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/STaSy/utils.py -------------------------------------------------------------------------------- /TabDDPM/CONFIG_DESCRIPTION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/CONFIG_DESCRIPTION.md -------------------------------------------------------------------------------- /TabDDPM/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/LICENSE.md -------------------------------------------------------------------------------- /TabDDPM/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/README.md -------------------------------------------------------------------------------- /TabDDPM/agg_results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/agg_results.ipynb -------------------------------------------------------------------------------- /TabDDPM/config/config.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/config/config.toml -------------------------------------------------------------------------------- /TabDDPM/config/config_test.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/config/config_test.toml -------------------------------------------------------------------------------- /TabDDPM/lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/__init__.py -------------------------------------------------------------------------------- /TabDDPM/lib/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/data.py -------------------------------------------------------------------------------- /TabDDPM/lib/dataset_prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/dataset_prep.py -------------------------------------------------------------------------------- /TabDDPM/lib/deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/deep.py -------------------------------------------------------------------------------- /TabDDPM/lib/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/env.py -------------------------------------------------------------------------------- /TabDDPM/lib/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/metrics.py -------------------------------------------------------------------------------- /TabDDPM/lib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/lib/util.py -------------------------------------------------------------------------------- /TabDDPM/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/requirements.txt -------------------------------------------------------------------------------- /TabDDPM/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /TabDDPM/scripts/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/scripts/pipeline.py -------------------------------------------------------------------------------- /TabDDPM/scripts/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/scripts/sample.py -------------------------------------------------------------------------------- /TabDDPM/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/scripts/train.py -------------------------------------------------------------------------------- /TabDDPM/scripts/utils_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/scripts/utils_train.py -------------------------------------------------------------------------------- /TabDDPM/tab_ddpm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/tab_ddpm/__init__.py -------------------------------------------------------------------------------- /TabDDPM/tab_ddpm/gaussian_multinomial_diffsuion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/tab_ddpm/gaussian_multinomial_diffsuion.py -------------------------------------------------------------------------------- /TabDDPM/tab_ddpm/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/tab_ddpm/modules.py -------------------------------------------------------------------------------- /TabDDPM/tab_ddpm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/TabDDPM/tab_ddpm/utils.py -------------------------------------------------------------------------------- /data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/data_loaders.py -------------------------------------------------------------------------------- /download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/download_data.py -------------------------------------------------------------------------------- /fi_shap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/fi_shap.png -------------------------------------------------------------------------------- /gain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/gain.py -------------------------------------------------------------------------------- /imputers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/imputers.py -------------------------------------------------------------------------------- /iris_cropped.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/iris_cropped.png -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/metrics.py -------------------------------------------------------------------------------- /missforest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/missforest.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /script_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/script_generation.py -------------------------------------------------------------------------------- /script_imputation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/script_imputation.py -------------------------------------------------------------------------------- /softimpute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/softimpute.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SamsungSAILMontreal/ForestDiffusion/HEAD/utils.py --------------------------------------------------------------------------------