Achille/samples - Githubissues

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

Closed ANazaret closed 3 months ago

ANazaret commented 3 months ago

Updates for the Samples class and the README

Remove subclassing of np.ndarray
Make Treeffuser.samples return np.ndarray again
Ensure that subscripting samples do not remove the sample or the batch dimension.
add a link to the paper
add the citation of the paper