grf icon indicating copy to clipboard operation
grf copied to clipboard

Access to value of criterion for splits

Open n-ihlo opened this issue 1 year ago • 1 comments
trafficstars

I'm interested in the structure of Causal Forests, and especially in the criteria used to select the splits. With get_tree I was able to access the structure, but this does not seem to include this criterion (or something similar like an impurity). Is it possible to get the value of the criterion for a split?

n-ihlo avatar May 16 '24 18:05 n-ihlo

Hi @n-ihlo, grf doesn't store the value of the criterion it uses to determine splits. Causal forest does CART splits on an appropriate "pseudo-outcome". All the information needed to reconstruct this pseudo-outcome is available in a fit forest. If you want to learn more about the criterion causal forest use, you could have a look at the grf paper which explains how CART splitting can be paired with influence functions to target heterogeneity in a variety of statistical parameters, such as treatment effects.

erikcs avatar May 19 '24 03:05 erikcs