Support for pandas dataframe

blei-lab / treeffuser

Treeffuser is an easy-to-use package for probabilistic prediction on tabular data with tree-based diffusion models.

https://blei-lab.github.io/treeffuser/

MIT License

14 stars 3 forks source link

Open ANazaret opened 7 hours ago

ANazaret commented 7 hours ago

Support pandas Dataframe as input to Treeffuser. In particular, users won't have to manually specific the categorical columns (with cat_idx).

Proposed in: #86