Prediction function for Universal Parametric Wrapper — predict.surv.parametric • SuperSurv

Obtains predicted survivals from a fitted surv.parametric object using exact closed-form equations.

Usage

# S3 method for class 'surv.parametric'
predict(object, newdata, new.times, ...)

Arguments

object: Fitted surv.parametric object.
newdata: New covariate data.frame.
new.times: Times at which to obtain predicted survivals.
...: Additional ignored arguments.

Value

A numeric matrix of predicted survival probabilities, where rows correspond to the observations in newdata and columns correspond to the evaluation times in new.times.

Examples

data("metabric", package = "SuperSurv")
dat <- metabric[1:30, ]
x_cols <- grep("^x", names(dat))[1:3]
X <- dat[, x_cols, drop = FALSE]
newX <- X[1:5, , drop = FALSE]
times <- seq(50, 150, by = 50)

fit <- surv.parametric(
  time = dat$duration,
  event = dat$event,
  X = X,
  newdata = newX,
  new.times = times,
  obsWeights = rep(1, nrow(dat)),
  id = NULL,
  dist = "weibull"
)

pred <- predict(fit$fit, newdata = newX, new.times = times)
dim(pred)
#> [1] 5 3