Adding Forest-Flow: Generating and Imputing Tabular Data via Diffusion and Flow-based Gradient-Boosted Trees

kilianFatras commented 1 year ago

What does this PR do?

This PR adds a notebook on the Forest-Flow method to TorchCFM. Forest-Flow's purpose is to generate tabular data with Flow Matching methods! We have added a notebook showing how to use XGBoost to train the vector field of I-CFM and generate tabular data. This has required the addition of a parameter 't' within the 'sample_location_and_conditional_flow' function within each class. As we have modified classes, we have also added our first test function.

Before submitting

[x] Did you make sure title is self-explanatory and the description concisely explains the PR?
[x] Did you make sure your PR does only one thing, instead of bundling different changes together?
[x] Did you list all the breaking changes introduced by this pull request?
[x] Did you test your PR locally with pytest command?
[x] Did you run pre-commit hooks with pre-commit run -a command?

Did you have fun?

Make sure you had fun coding 🙃

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

codecov-commenter commented 1 year ago

Codecov Report

Attention: 2 lines in your changes are missing coverage. Please review.

Comparison is base (81fcb8d) 35.63% compared to head (cac49f1) 35.81%.

Files	Patch %	Lines
torchcfm/conditional_flow_matching.py	83.33%	2 Missing :warning:

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #69 +/- ## ========================================== + Coverage 35.63% 35.81% +0.17% ========================================== Files 67 67 Lines 7417 7419 +2 ========================================== + Hits 2643 2657 +14 + Misses 4774 4762 -12 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

kilianFatras commented 1 year ago

@atong01 it seems that only the tests in runner are run and not the new one. Can you check that, please? Thank you

guillaumehu commented 1 year ago

@kilianFatras I had a similar issue, the directory name should be tests instead of test. Hope it works!

atong01 / conditional-flow-matching