Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Access to value of criterion for splits #1414

Open
n-ihlo opened this issue May 16, 2024 · 1 comment
Open

Access to value of criterion for splits #1414

n-ihlo opened this issue May 16, 2024 · 1 comment
Labels

Comments

@n-ihlo
Copy link

n-ihlo commented May 16, 2024

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?

@erikcs
Copy link
Member

erikcs commented May 19, 2024

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 erikcs added the question label Jun 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

2 participants