DR-VAE

pytorch code for discriminatively regularized variational autoencoders

Code to accompany paper

Discriminative Regularization for Latent Variable Models with Applications to Electrocardiography
Andrew Miller, Ziad Obermeyer, John Cunningham, Sendhil Mullainathan
International Conference on Machine Learning (ICML), 2019