CEVAE - pytorch

This repository contains a Python3, PyTorch implemenation of the Causal Effect Variational Autoencoder (CEVAE) model as developed at [1]. The original Python2 TensorFlow (+Edward) implementation can be found here: https://github.com/AMLab-Amsterdam/CEVAE

References

[1] Causal Effect Inference with Deep Latent-Variable Models Christos Louizos, Uri Shalit, Joris Mooij, David Sontag, Richard Zemel, Max Welling, 2017