Team-TUD / CTAB-GAN-Plus

Official GitHub for CTAB-GAN+
71 stars 10 forks source link

How to train and generate samples given a class/categorical variable? #10

Closed shrave closed 9 months ago

shrave commented 1 year ago

How to train and generate samples given a class/categorical variable? Like given a class, how can I control the generation to produce n samples for a particular class i.e. in this case a category in a categorical variable?

zhao-zilong commented 1 year ago

Hi @shrave sorry for the late response, so you can construct the condvec by yourself insteand of sample it here: https://github.com/Team-TUD/CTAB-GAN-Plus/blob/6d72fda3a9f382339e55cb4b35befced4c1f3508/model/synthesizer/ctabgan_synthesizer.py#L562 Then you can indicate any class as you want to give it as condition