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

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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