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
30 stars 6 forks source link

add Tree.values #58

Closed crflynn closed 3 years ago

crflynn commented 3 years ago

Calculates node_values at fit time rather than at access time. Compared to #55 this adds time to training but keeps the size of the serialized forest small, since we calculate the node values rather than store all the y-values and weights required to build them in Tree.