-
Notifications
You must be signed in to change notification settings - Fork 733
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
Vasilis/policy #377
Vasilis/policy #377
Conversation
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.
Looked mainly at the policy part and the notebooks. Everything looks good, some minor comments and one metadata fix required.
econml/policy/forest/_forest.py
Outdated
# ============================================================================= | ||
|
||
|
||
class PolicyForest(BaseEnsemble, metaclass=ABCMeta): |
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.
Same here, a lot of similarities with the GRF forest, except that the GRF has a lot more auxiliary functions (about 2/3 of the code). Most of the components here appear in the GRF forest as well.
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.
I tried for this one, but there was so much different small details that I thought it was ok for the forest.
I merged them in the tree, but left the forests separate.
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! Just comment on a few places need to add/update docstring.
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.
Mostly looks good, added a few small comments.
Shouldn't the new policy files also be added to our documentation? |
…ed notebooks. changed formating of text in tree plotting
done! Added the autosummary for the policy moduel |
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. I added a minor suggestion on the test.
…mes when none is passed in drlearner
… to non-fixed-randomness behavior
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.
I go over the docstring, looks great! Only in a few places miss max_depth
.
Added policy module.
Enabled policy interpreter for multiple treatments using a custom policy tree.
Add doubly robust forest policy learner.