-
Notifications
You must be signed in to change notification settings - Fork 792
add arg to allow missing values in W and sometimes X #791
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
Conversation
e07c5c2 to
ef790e5
Compare
kbattocchi
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good, but please add at least one test exercising this functionality.
90f6fff to
00dd506
Compare
kbattocchi
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks great.
|
Greetings! Per the office hours call - may we let the user opt in to nan's for X in the non parametric DML, the doubly robust learner, S learner, T learner and X learner? All of these models potentially allow for a CATE model able to handle nan's. |
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
9812ef4 to
e34e85f
Compare
…at dmlorf Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
kbattocchi
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
These updated changes look good to me.
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
Signed-off-by: Fabio Vera <[email protected]>
* enable nans in W Signed-off-by: Fabio Vera <[email protected]> * linting Signed-off-by: Fabio Vera <[email protected]> * add tests for ests that handle missing in W Signed-off-by: Fabio Vera <[email protected]> * allow missing in X for some ortholearner subclasses Signed-off-by: Fabio Vera <[email protected]> * refactor keyword arg to be bool only, add more tests Signed-off-by: Fabio Vera <[email protected]> * linting Signed-off-by: Fabio Vera <[email protected]> * enable missing for metalearners and orf, fix dowhywrapped discretetreat dmlorf Signed-off-by: Fabio Vera <[email protected]> * update arg name to allow_missing, add docstrings Signed-off-by: Fabio Vera <[email protected]> * add warning when missing values detected Signed-off-by: Fabio Vera <[email protected]> * dummy commit Signed-off-by: Fabio Vera <[email protected]> * dummy commit revert Signed-off-by: Fabio Vera <[email protected]> --------- Signed-off-by: Fabio Vera <[email protected]>
Summary of changes:
_OrthoLearner and subclasses
Metalearners
OrthoForest models