├── .gitignore ├── LICENSE ├── README.md ├── data └── Info │ ├── adult.json │ ├── adult_dcr.json │ ├── beijing.json │ ├── beijing_dcr.json │ ├── default.json │ ├── default_dcr.json │ ├── diabetes.json │ ├── diabetes_dcr.json │ ├── magic.json │ ├── news.json │ ├── news_dcr.json │ ├── news_nocat.json │ ├── shoppers.json │ └── shoppers_dcr.json ├── download_dataset.py ├── eval ├── eval_quality.py ├── mle │ ├── mle.py │ ├── tabular_dataload.py │ └── tabular_transformer.py └── visualize_density.py ├── eval_impute.py ├── images ├── tabdiff_demo.gif ├── tabdiff_demo.mp4 └── tabdiff_flowchart.jpg ├── main.py ├── process_dataset.py ├── src ├── __init__.py ├── data.py ├── env.py ├── metrics.py └── util.py ├── synthcity.yaml ├── tabdiff.yaml ├── tabdiff ├── configs │ └── tabdiff_configs.toml ├── main.py ├── metrics.py ├── models │ ├── noise_schedule.py │ └── unified_ctime_diffusion.py ├── modules │ ├── main_modules.py │ └── transformer.py └── trainer.py └── utils_train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/README.md -------------------------------------------------------------------------------- /data/Info/adult.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/adult.json -------------------------------------------------------------------------------- /data/Info/adult_dcr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/adult_dcr.json -------------------------------------------------------------------------------- /data/Info/beijing.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/beijing.json -------------------------------------------------------------------------------- /data/Info/beijing_dcr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/beijing_dcr.json -------------------------------------------------------------------------------- /data/Info/default.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/default.json -------------------------------------------------------------------------------- /data/Info/default_dcr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/default_dcr.json -------------------------------------------------------------------------------- /data/Info/diabetes.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/diabetes.json -------------------------------------------------------------------------------- /data/Info/diabetes_dcr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/diabetes_dcr.json -------------------------------------------------------------------------------- /data/Info/magic.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/magic.json -------------------------------------------------------------------------------- /data/Info/news.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/news.json -------------------------------------------------------------------------------- /data/Info/news_dcr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/news_dcr.json -------------------------------------------------------------------------------- /data/Info/news_nocat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/news_nocat.json -------------------------------------------------------------------------------- /data/Info/shoppers.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/shoppers.json -------------------------------------------------------------------------------- /data/Info/shoppers_dcr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/data/Info/shoppers_dcr.json -------------------------------------------------------------------------------- /download_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/download_dataset.py -------------------------------------------------------------------------------- /eval/eval_quality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/eval/eval_quality.py -------------------------------------------------------------------------------- /eval/mle/mle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/eval/mle/mle.py -------------------------------------------------------------------------------- /eval/mle/tabular_dataload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/eval/mle/tabular_dataload.py -------------------------------------------------------------------------------- /eval/mle/tabular_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/eval/mle/tabular_transformer.py -------------------------------------------------------------------------------- /eval/visualize_density.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/eval/visualize_density.py -------------------------------------------------------------------------------- /eval_impute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/eval_impute.py -------------------------------------------------------------------------------- /images/tabdiff_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/images/tabdiff_demo.gif -------------------------------------------------------------------------------- /images/tabdiff_demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/images/tabdiff_demo.mp4 -------------------------------------------------------------------------------- /images/tabdiff_flowchart.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/images/tabdiff_flowchart.jpg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/main.py -------------------------------------------------------------------------------- /process_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/process_dataset.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/src/data.py -------------------------------------------------------------------------------- /src/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/src/env.py -------------------------------------------------------------------------------- /src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/src/metrics.py -------------------------------------------------------------------------------- /src/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/src/util.py -------------------------------------------------------------------------------- /synthcity.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/synthcity.yaml -------------------------------------------------------------------------------- /tabdiff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff.yaml -------------------------------------------------------------------------------- /tabdiff/configs/tabdiff_configs.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/configs/tabdiff_configs.toml -------------------------------------------------------------------------------- /tabdiff/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/main.py -------------------------------------------------------------------------------- /tabdiff/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/metrics.py -------------------------------------------------------------------------------- /tabdiff/models/noise_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/models/noise_schedule.py -------------------------------------------------------------------------------- /tabdiff/models/unified_ctime_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/models/unified_ctime_diffusion.py -------------------------------------------------------------------------------- /tabdiff/modules/main_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/modules/main_modules.py -------------------------------------------------------------------------------- /tabdiff/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/modules/transformer.py -------------------------------------------------------------------------------- /tabdiff/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/tabdiff/trainer.py -------------------------------------------------------------------------------- /utils_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinkaiXu/TabDiff/HEAD/utils_train.py --------------------------------------------------------------------------------