ttgump / zinbae Goto Github PK

Hi,

Thanks for presenting this nice approach. I wonder where the gumble_softmax() function is used in ZINBAE.py. I could only find that the function is defined in line 97 but could not find where the output of this function is used to perform element-wise multiplication with the reconstructed mean estimated in the ZINB model. I wonder whether it is because I miss any important part of the code?

Looking forward to your response!

Thanks.