Learn R Programming
⚠️There's a newer version (1.2.3) of this package.Take me there.

policytree (version 1.0.1)

Policy Learning via Doubly Robust Empirical Welfare Maximization over Trees

Description

Learn optimal policies via doubly robust empirical welfare maximization over trees. This package implements the multi-action doubly robust approach of Zhou, Athey and Wager (2018) in the case where we want to learn policies that belong to the class of depth k decision trees.

Copy Link

Version

Install

install.packages('policytree')

Monthly Downloads

603

Version

1.0.1

License

GPL-3

Issues

14

Pull Requests

0

Stars

80

Forks

14

Maintainer

Erik Sverdrup

Last Published

July 13th, 2020

Functions in policytree (1.0.1)

conditional_means.causal_forest

Estimate mean rewards \(\mu\) for each treatment \(a\)
predict.policy_tree

Predict method for policy_tree
double_robust_scores.causal_forest

Matrix \(\Gamma\) of scores for each treatment \(a\)
print.multi_causal_forest

Print a multi_causal_forest object.
export_graphviz

Export a tree in DOT format. This function generates a GraphViz representation of the tree, which is then written into dot_string.
gen_data_epl

Example data generating process from Efficient Policy Learning
find_leaf_node

Query a tree with a sample
policy_tree

Fit a policy with exact tree search
plot.policy_tree

Plot a policy_tree tree object.
create_dot_body

Writes each node information If it is a leaf node: show it in different color, show number of samples, show leaf id If it is a non-leaf node: show its splitting variable and splitting value
gen_data_mapl

Example data generating process from Offline Multi-Action Policy Learning: Generalization and Optimization
multi_causal_forest

One vs. all causal forest for multiple treatment effect estimation
predict.multi_causal_forest

Predict with multi_causal_forest
print.policy_tree

Print a policy_tree object.
policytree-package

policytree: Policy Learning via Doubly Robust Empirical Welfare Maximization over Trees