Skip to content

Commit a496a67

Browse files
committed
fix: removed data.frame conversion of explainer in kernelshap
1 parent 40a076e commit a496a67

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

R/surv_shap.R

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -250,20 +250,14 @@ use_kernelshap <- function(explainer, new_observation, output_type, observation_
250250
"new_observation must be a data.frame" = inherits(
251251
new_observation, "data.frame")
252252
)
253-
# get explainer data to be able to make class checks and transformations
254-
explainer_data <- explainer$data
255-
# ensure that classes of explainer$data and new_observation are equal
256-
if (!inherits(explainer_data, "data.frame")) {
257-
explainer_data <- data.frame(explainer_data)
258-
}
259253

260254
shap_values <- sapply(
261255
X = as.character(seq_len(nrow(new_observation))),
262256
FUN = function(i) {
263257
tmp_res <- kernelshap::kernelshap(
264258
object = explainer$model,
265259
X = as.matrix(new_observation[as.integer(i), ]),
266-
bg_X = explainer$data,
260+
bg_X = as.matrix(explainer$data),
267261
pred_fun = predfun,
268262
verbose = FALSE
269263
)

0 commit comments

Comments
 (0)