search

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

Add support for categorical variable #58

Closed aagrande closed 5 months ago

aagrande commented 5 months ago

It closes #11 by adding support for categorical variables in both the main package and the testbed.

velezbeltran commented 5 months ago

@aagrande could you add a test that checks the categorical works and that if we don't use it the behavior of treefusser is the same