Skip to content

Code for TEDVAE, a VAE-based treatment effect estimation algorithm.

Notifications You must be signed in to change notification settings

WeijiaZhang24/TEDVAE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

a38ce70 · Dec 6, 2022

History

9 Commits
Jul 14, 2020
Jul 14, 2020
Jul 14, 2020
Dec 6, 2022
Jul 14, 2020
Jul 14, 2020
Jul 14, 2020

Repository files navigation

TEDVAE

Pytorch implementation for TEDVAE.

Weijia Zhang, Lin Liu, Jiuyong Li. (2021). Treatment Effect Estimation with Disentangled Latent Factors. Proceedings of the AAAI Conference on Artificial Intelligence, 35(12), 10923-10930. https://ojs.aaai.org/index.php/AAAI/article/view/17304

For enquiries, please send email to: weijia.zhang.xh@gmail.com

Requirement: Pytorch 1.5.0. Pyro 1.3.1.

The IHDP datasets are generated using https://github.com/vdorie/npci The ACIC datasets are generated using https://github.com/vdorie/aciccomp

About

Code for TEDVAE, a VAE-based treatment effect estimation algorithm.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages