search

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

Causal survival forest #41

Open crflynn opened 3 years ago

crflynn commented 3 years ago

implemented but not yet released in grf

erikcs commented 3 years ago

Just a heads up that this will be an annoying wrapper to make, since it involves a lot of tedious calculations from the wrapper side (sorry about that). It will probably be fine to postpone this for a while, since if Python users really want to try GRF for survival data they can always estimate Inverse Probability of Censoring Weights with a survival forest and pass those as sample_weights to a causal forest.