zhao-zilong / Tabula

Official git for "TabuLa: Harnessing Language Models for Tabular Data Synthesis"
31 stars 11 forks source link

Is it possible to make Tabula Conditional? #13

Closed HatedFate closed 2 weeks ago

HatedFate commented 1 month ago

If I want to generate samples with a specific feature, is there any ways to do this in Tabula? For example, I want to sample 100 data points of people with the name "John Doe".

zhao-zilong commented 1 month ago

@HatedFate yes, it is. Check this part of code: https://github.com/zhao-zilong/Tabula/blob/29cafbdb97c7fb61aa3318f7096f1e79e436a189/tabula/tabula.py#L234