Change base tree implementation to use Indexes (#23)

tabacof · Aug 30, 2023 · b7b2b72 · b7b2b72
1 parent 1ac556a
commit b7b2b72
Show file tree

Hide file tree

Showing 3 changed files with 208 additions and 283 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -7,6 +7,8 @@ mod utils;
 pub use dataset::Dataset;
 pub use trees::DecisionTree;
 pub use trees::RandomForest;
+pub use trees::TrainOptions;
+pub use trees::Tree;
 pub use utils::*;
 
 use pyo3::prelude::*;