py-why / py-why/EconML

The same heterogeneous treatment effect for each sample

Open
#438 6 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

Try to use NonParamDML for task with binary target Y~{0,1} and continious treatment T.

Define and fit model as

est3 = NonParamDML(model_y=CatBoostClassifier(n_estimators=10, max_depth=4),
                   model_t=CatBoostRegressor(n_estimators=10, max_depth=4),
                   model_final=RandomForestRegressor(),
                   discrete_treatment=False, 
                   cv=3)
est3.fit(train.target.values, train.treatment_size.values, X=train.loc[:, features_econml].values)

Then apply model to valid dataset:

te_pred_3 = est3.effect(valid.loc[:, features_econml].values) 

and get only one unique value in prediction: 8.35220653631326⋅10−6 and it is near to zero..

Feature values for each sample are different.

Tell me please, have I made any mistake?

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 by reproducing the shown NonParamDML fit and effect call with the stated models and checking the unique predictions. Compare the result with the expected behavior for differing feature values; done means determining whether the constant near-zero effect is expected or establishing a reproducible defect.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
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.