# NOT RUN {
library(StratifiedMedicine)
## Continuous ##
dat_ctns = generate_subgrp_data(family="gaussian")
Y = dat_ctns$Y
X = dat_ctns$X
A = dat_ctns$A
train = data.frame(Y, A, X)
## Estimate PLEs (ranger) ##
res_ranger = ple_train(Y, A, X, Xtest=X, ple = "ple_ranger")
## Identify Subgroups: MOB (lmtree) ##
res_lmtree = submod_train(Y, A, X, Xtest=X, submod="submod_lmtree")
## Parameter-estimation ##
params = param_ple(Y, A, X, mu_hat = res_ranger$mu_train,
Subgrps = res_lmtree$Subgrps.train, alpha_ovrl=0.05,
alpha_s=0.05)
params
# }
Run the code above in your browser using DataLab