show how to import the model

radiradev · May 6, 2024 · 323602f · 323602f
1 parent 60ba199
commit 323602f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -17,6 +17,7 @@ pip install flowmatching-bdt
 
 ```python
 from sklearn.datasets import make_moons
+from flowmatching_bdt import FlowMatchingBDT
 
 data, _ = make_moons(n_samples=1000, noise=0.1, random_state=0)
 model = FlowMatchingBDT()
@@ -35,6 +36,7 @@ If you'd like to do conditional generation:
 ```python
 import numpy as np
 from sklearn.datasets import make_moons
+from flowmatching_bdt import FlowMatchingBDT
 
 data, labels = make_moons(n_samples=1000, noise=0.1, random_state=42)
 model = FlowMatchingBDT()