Predict with a S-learner with Lasso

Obtain estimated tau(X) using a trained S-learner with Lasso model

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

Arguments

object

A surv_sl_lasso object

newdata

Covariate matrix to make predictions on. If null, return the tau(X) predictions on the training data

t0

The prediction time of interest

...

Additional arguments (currently not used)

Value

A vector of estimated conditional average treatment effects

Details

Remark: CATE predictions can be made at any time point on the estimated survival curve

Examples

# \donttest{
n <- 1000; p <- 25
t0 <- 0.2
Y.max <- 2
X <- matrix(rnorm(n * p), n, p)
W <- rbinom(n, 1, 0.5)
numeratorT <- -log(runif(n))
T <- (numeratorT / exp(1 * X[ ,1, drop = FALSE] + (-0.5 - 1 * X[ ,2, drop = FALSE]) * W)) ^ 2
failure.time <- pmin(T, Y.max)
numeratorC <- -log(runif(n))
censor.time <- (numeratorC / (4 ^ 2)) ^ (1 / 2)
Y <- as.numeric(pmin(failure.time, censor.time))
D <- as.integer(failure.time <= censor.time)
n.test <- 500
X.test <- matrix(rnorm(n.test * p), n.test, p)

surv.sl.lasso.fit <- surv_sl_lasso(X, Y, W, D, t0)
cate <- predict(surv.sl.lasso.fit)
cate.test <- predict(surv.sl.lasso.fit, X.test)
# }