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: [email protected]

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