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 node count, capacity, n outputs, n classes to tree #55

Closed crflynn closed 3 years ago

crflynn commented 3 years ago

Move Tree class to _tree module. Default setting n_classes_ to 1 instead of None.

Adds

to Tree.