Generative models and tabular data

[Chechgm]

Hi Yashu, I read your post and code on NN and tabular data and found your code to be super elegant. I am currently working with tabular data as well but for generative modeling (VAEs and WGANs, specifically) and I am looking into writing some code which allows me to come back from samples from these models to the original dataset structure. I have already written something but is super verbose and I am afraid that, at some point, it won't be maintenable. Do you know of any code, as elegant as from the post, that does this?