predict_rrt_reg: Compute the prediction of a model built with 'rrt_reg'

In YohannLeFaou/sword: Random forest with IPC weights for survival analysis

Description

Given a model built wtih rrt_reg, predict_rrt_reg allows to get the predictions of the model for new observations

Usage

predict_rrt_reg(obj, newdata, type)

Arguments

obj	A list output by rrt_reg
newdata	A data.frame which contains the same variables as the ones used for the training
type	A character string which gives the type of terminal leaf estimator to use. If "exp", IPCW used to grow the tree are used. If "KMloc", a Kaplan-Meier estimator is fitted in each terminal leaf

Value

A list with the following elements :

pred	The vector of the predicted values for phi(T') (with T' = min(T, max_time)) for the observations of newdata
surv	The matrix which contains the estimated values of the survival curves at time_points, for the observations of newdata
time_points	The vector of the time points where the survival curves are evaluated

See Also

rrt_reg

YohannLeFaou/sword documentation built on May 28, 2019, 3:21 p.m.