StratifiedMedicine (version 0.1.3)

predict.submod_lmtree: Predict submod: lmtree

Description

Predict subgroups and obtain subgroup-specific estimates of E(Y|A=1)-E(Y|A=0).

Usage

# S3 method for submod_lmtree
predict(object, newdata = NULL, ...)

Arguments

object

Trained lmtree model.

newdata

Data-set to make predictions at (Default=NULL, predictions correspond to training data).

...

Any additional parameters, not currently passed through.

Value

Identified subgroups with subgroup-specific predictions of E(Y|A=1)-E(Y|A=0).

• Subgrps - Identified subgroups

• pred - Predictions, E(Y|A=1)-E(Y|A=0) by subgroup.

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

res_lmtree1 = submod_lmtree(Y, A, X, Xtest=X)
# Predict subgroups / estimates #
out = predict(res_lmtree1)


# }