tunePHspline {survivalSL} | R Documentation |
Tune a Survival Regression using the Royston/Parmar Spline Model
Description
This function finds the optimal number of knots of the spline function.
Usage
tunePHspline(formula,
data, cv=10, metric="auc", k=1:4, pro.time=NULL,
seed=NULL, ROC.precision=seq(.01, .99, by=.01))
Arguments
formula |
A formula object, with the response on the left of a ~ operator, and the terms on the right. The response must be a survival object as returned by the |
data |
A data frame for training the model with the same covariates as in the formula. |
cv |
The value of the number of folds. The default value is 10. |
metric |
The loss function or metric. See details. Default metric is Area Under ROC ( |
k |
The number of knots optimized over. |
pro.time |
This optional value of prognostic time represents the maximum delay for which the capacity of the variable is evaluated. The same unit than the one used in the argument times. Not used for the following metrics: "ll", "ibs", and "ibll". Default value is the time at which half of the subjects are still at risk. |
seed |
A random seed to ensure reproducibility during the cv process. If |
ROC.precision |
The percentiles (between 0 and 1) of the prognostic variable used for computing each point of the time dependent ROC curve. Only used when |
Details
The function runs the flexsurvspline
function of the flexsurv
package. The following metrics can be used : "bs" for the Brier score at the prognostic time pro.time
, "p_ci" and "uno_ci" for the concordance index at the prognostic time pro.time
(Pencina and Uno versions), "ll" for the log-likelihood, "ibs" for the integrated Brier score up to the last observed time in the training data, "ibll" for the integrated binomial log-likelihood up to the last observed time in the training data, "ribs" for the restricted integrated Brier score up to the prognostic time pro.time
, "ribll" for the restricted integrated binomial log-likelihood up to the prognostic time pro.time
, "bll" for the binomial log-likelihood, "auc" for the area under the time-dependent ROC curve up to the prognostic time pro.time
.
Value
optimal |
The value of |
results |
The data frame with the cross-validated value of the metric/loss function according to |
References
Royston, P. and Parmar, M. (2002). Flexible parametric proportional-hazards and proportional odds models for censored survival data, with application to prognostic modelling and estimation of treatment effects. Statistics in Medicine 21(1):2175-2197. doi: 10.1002/sim.1203
Examples
data("dataDIVAT2")
# The estimation of the hyperparameters on the first 150 patients
formula<-Surv(times,failures) ~ age + hla + retransplant + ecd
tune.model <- tunePHspline(formula=formula,
data=dataDIVAT2[1:150,], cv=3, metric="auc",
k=1:2, pro.time=NULL,seed=123,
ROC.precision=seq(.01, .99, by=.02))
# the estimated nodesize value
tune.model$optimal
tune.model$results