search

kim-hyunsu / CEVAE-pyro

CEVAE(Causal Effect Variational AutoEncoder) written with pytorch and pyro.
9 stars 4 forks source link

readme

CEVAE-pyro

Pytorch and Pyro implementation of CEVAE(Causal Effect Variational AutoEncoder). See https://arxiv.org/pdf/1705.08821.pdf

Warning

This project is built without correcting bugs. It may produces poor computational time and accuracy.

Issue reporting is always welcome!

Run

python3 main.py [options]

See main.py to know the options.

Environment

References

  1. https://pyro.ai/examples/vae.html
  2. https://github.com/AMLab-Amsterdam/CEVAE
  3. https://github.com/belaalb/CEVAE-VampPrior