├── .bumpversion.cfg ├── .cookiecutterrc ├── .coveragerc ├── .github └── workflows │ └── github-actions.yml ├── .gitignore ├── .pre-commit-config.yaml ├── AUTHORS.md ├── CHANGELOG.rst ├── CONTRIBUTING.rst ├── LICENSE ├── Makefile ├── README.md ├── README_example.png ├── ci ├── bootstrap.py ├── requirements.txt └── templates │ └── .github │ └── workflows │ └── github-actions.yml ├── dev_tools ├── README.md └── freeze_requirements.sh ├── examples ├── README_example.ipynb ├── README_example.py └── m5-news-vendor-notebook.ipynb ├── pyproject.toml ├── pytest.ini ├── setup_deprecated.py ├── src └── treeffuser │ ├── __init__.py │ ├── _base_tabular_diffusion.py │ ├── _scaler.py │ ├── _score_models.py │ ├── _warnings.py │ ├── samples.py │ ├── sde │ ├── __init__.py │ ├── base_sde.py │ ├── base_solver.py │ ├── diffusion_sdes.py │ ├── initialize.py │ ├── parameter_schedule.py │ └── solvers.py │ └── treeffuser.py ├── testbed ├── Makefile ├── notebooks │ ├── card_example.py │ ├── example.py │ ├── m5-news-vendor-notebook.ipynb │ ├── m5-ppc-notebook.ipynb │ ├── paper_notebooks │ │ ├── benchmark_uci_et_al.py │ │ ├── inflated │ │ │ ├── _generator.py │ │ │ ├── fig │ │ │ │ ├── branching_mixture.pdf │ │ │ │ └── covariate_inflated.pdf │ │ │ └── main_inflated.ipynb │ │ ├── misc │ │ │ └── he.jpg │ │ ├── multimodality │ │ │ ├── _branching_mixture.py │ │ │ ├── fig │ │ │ │ └── branching_mixture.pdf │ │ │ └── main_branching_mixture.ipynb │ │ ├── multivariate-example.py │ │ └── speed_benchmark.py │ └── testbed_example.py ├── pyproject.toml ├── pytest.ini ├── scripts │ ├── submit_m5_subset.py │ ├── submit_slurm_job.py │ └── submit_slurm_job_synthetic.py ├── setup.py ├── src │ └── testbed │ │ ├── __init__.py │ │ ├── __main__.py │ │ ├── data │ │ ├── heavy_tails │ │ │ ├── m5_subset │ │ │ │ ├── preprocess.py │ │ │ │ └── raw │ │ │ │ │ └── dataset.pkl │ │ │ └── movies │ │ │ │ └── preprocess.py │ │ ├── links.json │ │ ├── m5.tar.gz │ │ ├── readme.md │ │ ├── uci │ │ │ ├── bike │ │ │ │ └── preprocess.py │ │ │ ├── communities │ │ │ │ └── preprocess.py │ │ │ ├── energy │ │ │ │ └── preprocess.py │ │ │ ├── facebook │ │ │ │ └── preprocess.py │ │ │ ├── kin8nm │ │ │ │ ├── preprocess.py │ │ │ │ └── raw │ │ │ │ │ └── data.zip │ │ │ ├── msd │ │ │ │ └── preprocess.py │ │ │ ├── naval │ │ │ │ └── preprocess.py │ │ │ ├── news │ │ │ │ └── preprocess.py │ │ │ ├── power │ │ │ │ └── preprocess.py │ │ │ ├── protein │ │ │ │ └── preprocess.py │ │ │ ├── superconductor │ │ │ │ └── preprocess.py │ │ │ ├── wave │ │ │ │ └── preprocess.py │ │ │ ├── wine │ │ │ │ └── preprocess.py │ │ │ └── yacht │ │ │ │ └── preprocess.py │ │ └── utils.py │ │ ├── distributions │ │ ├── __init__.py │ │ └── distribution.py │ │ ├── metrics │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── base_metric.py │ │ ├── calibration.py │ │ ├── crps.py │ │ └── log_likelihood.py │ │ ├── models │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── drf_.py │ │ ├── ibug_.py │ │ ├── lightning_uq_models │ │ │ ├── __init__.py │ │ │ ├── _card.py │ │ │ ├── _data_module.py │ │ │ ├── _deep_ensemble.py │ │ │ ├── _mc_dropout.py │ │ │ ├── _quantile_regression.py │ │ │ └── _utils.py │ │ ├── ngboost │ │ │ ├── __init__.py │ │ │ ├── _gaussian_mixtures.py │ │ │ └── _ngboost.py │ │ ├── nnffuser │ │ │ ├── __init__.py │ │ │ ├── _nnfuser.py │ │ │ └── _score.py │ │ ├── point_prediction_model.py │ │ ├── quantile_regression.py │ │ └── treeffuser.py │ │ ├── run_simulated_datasets.py │ │ └── simulated_data │ │ ├── __init__.py │ │ ├── base.py │ │ └── continuous.py └── tests │ ├── __init__.py │ ├── models │ ├── test_base_model.py │ ├── test_deep_learning_models.py │ └── test_quantile_regression.py │ └── simulated_data │ └── test_continuous.py ├── tests ├── __init__.py ├── sde │ ├── __init__.py │ ├── test_base_sde.py │ ├── test_initialize.py │ ├── test_sdes_and_solvers.py │ └── test_solvers.py ├── test_samples.py ├── test_score_models.py ├── test_treeffuser.py └── utils.py └── tox.ini /.bumpversion.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/.bumpversion.cfg -------------------------------------------------------------------------------- /.cookiecutterrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/.cookiecutterrc -------------------------------------------------------------------------------- /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/.coveragerc -------------------------------------------------------------------------------- /.github/workflows/github-actions.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/.github/workflows/github-actions.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /AUTHORS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/AUTHORS.md -------------------------------------------------------------------------------- /CHANGELOG.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/CHANGELOG.rst -------------------------------------------------------------------------------- /CONTRIBUTING.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/CONTRIBUTING.rst -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/README.md -------------------------------------------------------------------------------- /README_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/README_example.png -------------------------------------------------------------------------------- /ci/bootstrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/ci/bootstrap.py -------------------------------------------------------------------------------- /ci/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/ci/requirements.txt -------------------------------------------------------------------------------- /ci/templates/.github/workflows/github-actions.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/ci/templates/.github/workflows/github-actions.yml -------------------------------------------------------------------------------- /dev_tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/dev_tools/README.md -------------------------------------------------------------------------------- /dev_tools/freeze_requirements.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/dev_tools/freeze_requirements.sh -------------------------------------------------------------------------------- /examples/README_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/examples/README_example.ipynb -------------------------------------------------------------------------------- /examples/README_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/examples/README_example.py -------------------------------------------------------------------------------- /examples/m5-news-vendor-notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/examples/m5-news-vendor-notebook.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/pytest.ini -------------------------------------------------------------------------------- /setup_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/setup_deprecated.py -------------------------------------------------------------------------------- /src/treeffuser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/__init__.py -------------------------------------------------------------------------------- /src/treeffuser/_base_tabular_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/_base_tabular_diffusion.py -------------------------------------------------------------------------------- /src/treeffuser/_scaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/_scaler.py -------------------------------------------------------------------------------- /src/treeffuser/_score_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/_score_models.py -------------------------------------------------------------------------------- /src/treeffuser/_warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/_warnings.py -------------------------------------------------------------------------------- /src/treeffuser/samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/samples.py -------------------------------------------------------------------------------- /src/treeffuser/sde/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/__init__.py -------------------------------------------------------------------------------- /src/treeffuser/sde/base_sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/base_sde.py -------------------------------------------------------------------------------- /src/treeffuser/sde/base_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/base_solver.py -------------------------------------------------------------------------------- /src/treeffuser/sde/diffusion_sdes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/diffusion_sdes.py -------------------------------------------------------------------------------- /src/treeffuser/sde/initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/initialize.py -------------------------------------------------------------------------------- /src/treeffuser/sde/parameter_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/parameter_schedule.py -------------------------------------------------------------------------------- /src/treeffuser/sde/solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/sde/solvers.py -------------------------------------------------------------------------------- /src/treeffuser/treeffuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/src/treeffuser/treeffuser.py -------------------------------------------------------------------------------- /testbed/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/Makefile -------------------------------------------------------------------------------- /testbed/notebooks/card_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/card_example.py -------------------------------------------------------------------------------- /testbed/notebooks/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/example.py -------------------------------------------------------------------------------- /testbed/notebooks/m5-news-vendor-notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/m5-news-vendor-notebook.ipynb -------------------------------------------------------------------------------- /testbed/notebooks/m5-ppc-notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/m5-ppc-notebook.ipynb -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/benchmark_uci_et_al.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/benchmark_uci_et_al.py -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/inflated/_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/inflated/_generator.py -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/inflated/fig/branching_mixture.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/inflated/fig/branching_mixture.pdf -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/inflated/fig/covariate_inflated.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/inflated/fig/covariate_inflated.pdf -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/inflated/main_inflated.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/inflated/main_inflated.ipynb -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/misc/he.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/misc/he.jpg -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/multimodality/_branching_mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/multimodality/_branching_mixture.py -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/multimodality/fig/branching_mixture.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/multimodality/fig/branching_mixture.pdf -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/multimodality/main_branching_mixture.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/multimodality/main_branching_mixture.ipynb -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/multivariate-example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/multivariate-example.py -------------------------------------------------------------------------------- /testbed/notebooks/paper_notebooks/speed_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/paper_notebooks/speed_benchmark.py -------------------------------------------------------------------------------- /testbed/notebooks/testbed_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/notebooks/testbed_example.py -------------------------------------------------------------------------------- /testbed/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/pyproject.toml -------------------------------------------------------------------------------- /testbed/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/pytest.ini -------------------------------------------------------------------------------- /testbed/scripts/submit_m5_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/scripts/submit_m5_subset.py -------------------------------------------------------------------------------- /testbed/scripts/submit_slurm_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/scripts/submit_slurm_job.py -------------------------------------------------------------------------------- /testbed/scripts/submit_slurm_job_synthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/scripts/submit_slurm_job_synthetic.py -------------------------------------------------------------------------------- /testbed/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/setup.py -------------------------------------------------------------------------------- /testbed/src/testbed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/__main__.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/heavy_tails/m5_subset/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/heavy_tails/m5_subset/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/heavy_tails/m5_subset/raw/dataset.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/heavy_tails/m5_subset/raw/dataset.pkl -------------------------------------------------------------------------------- /testbed/src/testbed/data/heavy_tails/movies/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/heavy_tails/movies/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/links.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/links.json -------------------------------------------------------------------------------- /testbed/src/testbed/data/m5.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/m5.tar.gz -------------------------------------------------------------------------------- /testbed/src/testbed/data/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/readme.md -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/bike/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/bike/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/communities/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/communities/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/energy/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/energy/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/facebook/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/facebook/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/kin8nm/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/kin8nm/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/kin8nm/raw/data.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/kin8nm/raw/data.zip -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/msd/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/msd/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/naval/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/naval/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/news/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/news/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/power/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/power/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/protein/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/protein/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/superconductor/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/superconductor/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/wave/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/wave/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/wine/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/wine/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/uci/yacht/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/uci/yacht/preprocess.py -------------------------------------------------------------------------------- /testbed/src/testbed/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/data/utils.py -------------------------------------------------------------------------------- /testbed/src/testbed/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/distributions/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/distributions/distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/distributions/distribution.py -------------------------------------------------------------------------------- /testbed/src/testbed/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/metrics/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/metrics/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/metrics/accuracy.py -------------------------------------------------------------------------------- /testbed/src/testbed/metrics/base_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/metrics/base_metric.py -------------------------------------------------------------------------------- /testbed/src/testbed/metrics/calibration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/metrics/calibration.py -------------------------------------------------------------------------------- /testbed/src/testbed/metrics/crps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/metrics/crps.py -------------------------------------------------------------------------------- /testbed/src/testbed/metrics/log_likelihood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/metrics/log_likelihood.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/base_model.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/drf_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/drf_.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/ibug_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/ibug_.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/_card.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/_card.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/_data_module.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/_deep_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/_deep_ensemble.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/_mc_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/_mc_dropout.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/_quantile_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/_quantile_regression.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/lightning_uq_models/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/lightning_uq_models/_utils.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/ngboost/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/ngboost/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/ngboost/_gaussian_mixtures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/ngboost/_gaussian_mixtures.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/ngboost/_ngboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/ngboost/_ngboost.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/nnffuser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/nnffuser/__init__.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/nnffuser/_nnfuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/nnffuser/_nnfuser.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/nnffuser/_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/nnffuser/_score.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/point_prediction_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/point_prediction_model.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/quantile_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/quantile_regression.py -------------------------------------------------------------------------------- /testbed/src/testbed/models/treeffuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/models/treeffuser.py -------------------------------------------------------------------------------- /testbed/src/testbed/run_simulated_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/run_simulated_datasets.py -------------------------------------------------------------------------------- /testbed/src/testbed/simulated_data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /testbed/src/testbed/simulated_data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/simulated_data/base.py -------------------------------------------------------------------------------- /testbed/src/testbed/simulated_data/continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/src/testbed/simulated_data/continuous.py -------------------------------------------------------------------------------- /testbed/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /testbed/tests/models/test_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/tests/models/test_base_model.py -------------------------------------------------------------------------------- /testbed/tests/models/test_deep_learning_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/tests/models/test_deep_learning_models.py -------------------------------------------------------------------------------- /testbed/tests/models/test_quantile_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/tests/models/test_quantile_regression.py -------------------------------------------------------------------------------- /testbed/tests/simulated_data/test_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/testbed/tests/simulated_data/test_continuous.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/sde/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/sde/test_base_sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/sde/test_base_sde.py -------------------------------------------------------------------------------- /tests/sde/test_initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/sde/test_initialize.py -------------------------------------------------------------------------------- /tests/sde/test_sdes_and_solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/sde/test_sdes_and_solvers.py -------------------------------------------------------------------------------- /tests/sde/test_solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/sde/test_solvers.py -------------------------------------------------------------------------------- /tests/test_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/test_samples.py -------------------------------------------------------------------------------- /tests/test_score_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/test_score_models.py -------------------------------------------------------------------------------- /tests/test_treeffuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/test_treeffuser.py -------------------------------------------------------------------------------- /tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tests/utils.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blei-lab/treeffuser/HEAD/tox.ini --------------------------------------------------------------------------------