infer_variable_type function

Extract information of selected variables from high-dimensional Cox models

Extract the names and type of selected variables from fitted high-dimensional Cox models.

infer_variable_type(object, x)

Arguments

• object: Model object.
• x: Data matrix used to fit the model.

Returns

A list containing the index, name, type and range of the selected variables.

Examples

data("smart")
x <- as.matrix(smart[, -c(1, 2)])
time <- smart$TEVENT
event <- smart$EVENT
y <- survival::Surv(time, event)

fit <- fit_lasso(x, y, nfolds = 5, rule = "lambda.1se", seed = 11)
infer_variable_type(fit, x)