A question about Loss

[vicscott]

First I have to say it is a nice work! I have noticed that you focused on the generative process in Section 2.1 where describing the NBVAE. And in the code, the KL loss means that the encoder was still sampling from Gaussian distribution. And this is my question, why not sample from X~NB(r,p)? And if features were learned from Gaussian in the encoder, how can the decoder reconstruct the information following NB distribution? Thank you!