Skip to content
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

Enable treatment assignment from SingleTreePolicyInterpreter #222

Closed
kbattocchi opened this issue Feb 18, 2020 · 0 comments · Fixed by #228
Closed

Enable treatment assignment from SingleTreePolicyInterpreter #222

kbattocchi opened this issue Feb 18, 2020 · 0 comments · Fixed by #228
Assignees
Labels
enhancement New feature or request

Comments

@kbattocchi
Copy link
Collaborator

Given a trained SingleTreePolicyInterpreter, it would be nice to be able to do:

T = stpi.assign_treatment(X)

to get the suggested treatments corresponding to units with the given features given the learned policy.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant