py-why / py-why/EconML

Ortholearner fitfinal does not get the expanded treatment

Open
#313 0 comments 0 reactions 3 assignees View on GitHub

@heimengqi is already working on this.

Since Nov 12, 2020.

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

Description

https://github.com/microsoft/EconML/blob/a79bdeabc451747ffefe74ab5b4c0dfd124d1a38/econml/_ortho_learner.py#L542

Currently the expanded treatment is passed only to the nuisance models but not the final model.

This has not affected any behavior so far as all our estimators dont use the raw treatment in the final stage ever. But for uniformity and potential future estimators we should adhere I believe to always passing the expanded treatment when discrete_treatment=True.

Most prob the treatment expansion snippet needs to move to the fit method not inside the fit_nuisance method.

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.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.