predict.submod_train: Subgroup Identification: Train Model (Predictions)

Description Usage Arguments Value Examples

View source: R/submod_train.R

Description

Prediction function for the trained subgroup identification model (submod).

Usage

1
2
## S3 method for class 'submod_train'
predict(object, newdata = NULL, ...)

Arguments

object

Trained submod 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 (depends on subgroup model)

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
library(StratifiedMedicine)
## Continuous ##
dat_ctns = generate_subgrp_data(family="gaussian")
Y = dat_ctns$Y
X = dat_ctns$X
A = dat_ctns$A

# Fit submod_lmtree directly #
mod1 = submod_lmtree(Y, A, X, Xtest=X)
plot(mod1$mod)

# Fit through submod_train wrapper #
mod2 = submod_train(Y=Y, A=A, X=X, Xtest=X, submod="submod_lmtree")
out2 = predict(mod2)
plot(mod2$fit$mod)

StratifiedMedicine documentation built on March 1, 2020, 9:07 a.m.