py-why / py-why/EconML

GRF Causal Forest alpha and pointJ question

Open
#752 8 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Jupyter Notebook
Stars
4.8k
Forks
827
PR merge metrics
No merged PRs in 30d

Description

Hi,

I was using the GRF code (https://github.com/py-why/EconML/blob/8b7fe338600b7ccb6b8362f658d0ec35f5c75b7a/econml/grf/classes.py#L394-L399) and I noticed that the alpha and pointJ for CATE estimation is defined as:

  • y * T
  • T x T (cross-product)

I presumed pointJ was the Jacobian being estimated. This brought up some confusion for me and now I was hoping to clarify with the dev team here:

  1. what is alpha in the GRF code and why is it y * T? What if T is a vector and y is just a univariate outcome? It's not documented anywhere what the behavior here is.
  2. why is the point-wise Jacobian of the moment equation just the cross-product of the treatment arrays?

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with econml/grf/classes.py at lines 394-399 and trace how alpha and pointJ are used in CATE estimation. Clarify the treatment-vector and univariate-outcome behavior and explain why the cross-product represents the point-wise Jacobian; document the resulting semantics in the relevant GRF documentation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Documentation
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.