crflynn / skgrf

scikit-learn compatible Python bindings for grf (generalized random forests) C++ random forest library
https://skgrf.readthedocs.io/en/stable/
GNU General Public License v3.0
31 stars 7 forks source link

add a low-level tree object similar to sklearn's #51

Closed crflynn closed 3 years ago

crflynn commented 3 years ago

Instead of returning self from tree_, create an actual Tree class to mimic the sklearn low-level tree interface