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

Scaling #59

Closed velezbeltran closed 5 months ago

velezbeltran commented 5 months ago

Fixes issues with scaling and I added some tests to make sure that the models work (due to the issues that @ANazaret was having).

ANazaret commented 5 months ago

The base branch was not main, these changes were not applied.