Team-Wang-Lab/T-RL: Tree-based Reinforcement Learning for estimating optimal DTR.

We propose a tree-based reinforcement learning (T-RL) method to directly estimate optimal DTRs in a multi-stage multi-treatment setting. At each stage, T-RL builds an unsupervised decision tree that directly handles the problem of optimization with multiple treatment comparisons, through a purity measure constructed with augmented inverse probability weighted estimators.

Getting started

Package details

AuthorTebin Tao, Nina Zhou, Lu Wang
MaintainerLu Wang <luwang@umich.edu>, Nina Zhou <zhounina@umich.edu>
LicenseGPL (>= 2)
Version0.1.0
Package repositoryView on GitHub
Installation Install the latest version of this package by entering the following in R:
install.packages("remotes")
remotes::install_github("Team-Wang-Lab/T-RL")
Team-Wang-Lab/T-RL documentation built on Jan. 3, 2020, 12:11 a.m.