-
Notifications
You must be signed in to change notification settings - Fork 763
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
code cleanups #634
code cleanups #634
Conversation
@jeongyoonlee updated! |
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.
Thanks for the contribution! LGTM except one for tmle.py
. If you'd like to keep this PR simple without touching tmle.py
, that's fine too.
causalml/inference/meta/tmle.py
Outdated
@@ -126,7 +126,7 @@ def __repr__(self): | |||
self.__class__.__name__, self.model_tau.__repr__(), self.cv | |||
) | |||
|
|||
def estimate_ate(self, X, treatment, y, p, segment=None, return_ci=False): |
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.
Ideally, it should follow the signature from causalml.inference.meta.base.BaseLearner.estimate_ate()
. Can you update TMLELearner
to inherit from BaseLearner
and keep the function signatures consistent with other meta learners?
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.
Updated @jeongyoonlee and will keep this pull request simple. Will open a new one for a deeper dive into tmle.py
Keeping this pull request simple.
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.
LGTM. Thanks!
* code cleanups * lint error
Proposed changes
Small code cleanups.
Types of changes
What types of changes does your code introduce to CausalML?
Put an
x
in the boxes that applyChecklist
Put an
x
in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your code.Further comments
If this is a relatively large or complex change, kick off the discussion by explaining why you chose the solution you did and what alternatives you considered, etc. This PR template is adopted from appium.