StratifiedMedicine (version 0.1.3)

submod_otr: Subgroup Identification: Optimal Treatment Regime (through ctree)

Description

For continuous, binary, or survival outcomes, regress I(PLE>thres)~X with weights=abs(PLE) in ctree.

Usage

submod_otr(Y, A, X, Xtest, mu_train, minbucket = floor(dim(X)[1] * 0.1),
  maxdepth = 4, thres = ">0", ...)

Arguments

Y

The outcome variable. Must be numeric or survival (ex; Surv(time,cens) )

A

Treatment variable. (a=1,...A)

X

Covariate space.

Xtest

Test set

mu_train

Patient-level estimates (See PLE_models)

minbucket

Minimum number of observations in a tree node. Default = floor( dim(train)[1]*0.05 )

maxdepth

Maximum depth of any node in the tree (default=4)

thres

Threshold for PLE, ex: I(PLE>thres). Default is ">0". Direction can be reversed and can include equality sign (ex: "<=")

...

Any additional parameters, not currently passed through.

Value

Trained ctree (optimal treatment regime) model.

• mod - tree (OTR) model object

Examples

# NOT RUN {
library(StratifiedMedicine)

## Continuous ##
dat_ctns = generate_subgrp_data(family="gaussian")
Y = dat_ctns$Y
X = dat_ctns$X
A = dat_ctns$A

# }
# NOT RUN {
## Estimate PLEs (through Ranger) ##
res.ple = ple_model(Y, A, X, Xtest=X, family="gaussian", ple="ple_ranger")

## Fit OTR Subgroup Model ##
res_otr = submod_otr(Y, A, X, Xtest=X, mu_train = res.ple$mu_train)
plot(res_otr$mod)
# }
# NOT RUN {
# }