View source: R/variable_response.R
variable_response | R Documentation |
Function variable_response
calculates the expected output condition on a selected variable.
variable_response(explainer, variable, type = "pdp", link = explainer$link)
explainer |
an object of the class 'surv_explainer'. |
variable |
a character with variable name. |
type |
a character - type of the response to be calculated. Currently following options are implemented: 'pdp' for Partial Dependency. |
link |
a function - a link function that shall be applied to raw model predictions. This will be inherited from the explainer. |
library(survxai) library(rms) data("pbcTest") data("pbcTrain") predict_times <- function(model, data, times){ prob <- rms::survest(model, data, times = times)$surv return(prob) } cph_model <- cph(Surv(years, status)~sex + bili + stage, data=pbcTrain, surv=TRUE, x = TRUE, y=TRUE) surve_cph <- explain(model = cph_model, data = pbcTest[,-c(1,5)], y = Surv(pbcTest$years, pbcTest$status), predict_function = predict_times) svr_cph <- variable_response(surve_cph, "sex")
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.