Skip to content

Commit 40a076e

Browse files
committed
fix: new_observation now as.matrix for kernelshap
1 parent 0e8c85c commit 40a076e

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

R/surv_shap.R

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -257,20 +257,12 @@ use_kernelshap <- function(explainer, new_observation, output_type, observation_
257257
explainer_data <- data.frame(explainer_data)
258258
}
259259

260-
predfun <- function(model, newdata){
261-
explainer$predict_survival_function(
262-
model,
263-
newdata,
264-
times = explainer$times
265-
)
266-
}
267-
268260
shap_values <- sapply(
269261
X = as.character(seq_len(nrow(new_observation))),
270262
FUN = function(i) {
271263
tmp_res <- kernelshap::kernelshap(
272264
object = explainer$model,
273-
X = new_observation[as.integer(i), ],
265+
X = as.matrix(new_observation[as.integer(i), ]),
274266
bg_X = explainer$data,
275267
pred_fun = predfun,
276268
verbose = FALSE

0 commit comments

Comments
 (0)