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