StratifiedMedicine (version 0.1.3)

predict.ple_train: Patient-level Estimates Model: Prediction

Description

Prediction function for the trained patient-level estimate (ple) model.

Usage

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

Arguments

object

Trained ple model.

newdata

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

...

Any additional parameters, not currently passed through.

Value

Data-frame with predictions (depends on trained ple model).

See Also

PRISM

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


# Fit through ple_train wrapper #
mod2 = ple_train(Y=Y, A=A, X=X, Xtest=X, ple="ple_ranger" )
summary(mod2$mu_train)

res2 = predict(mod2) # newdata=NULL, training data #
res3 = predict(mod2, newdata=X) # test data #
summary(res2)
summary(res3)
# }
# NOT RUN {
# }