enable to pass additional parameters to treeshap function

Author: krzyzinskim

R/model_survshap.R

@@ -102,7 +102,8 @@ model_survshap.surv_explainer <- function(explainer,
         N = N,
         y_true = y_true,
         calculation_method = calculation_method,
-        aggregation_method = aggregation_method
+        aggregation_method = aggregation_method,
+        ...
     )
 
     attr(shap_values, "label") <- explainer$label

R/surv_shap.R

@@ -252,14 +252,13 @@ use_kernelshap <- function(explainer, new_observation, output_type, N, ...) {
             new_observation, "data.frame")
     )
 
+    if (is.null(N)) N <- nrow(explainer$data)
     background_data <- explainer$data[sample(1:nrow(explainer$data), N),]
     # ensure that classes of explainer$data and new_observation are equal
     if (!inherits(background_data, "data.frame")) {
         background_data <- data.frame(background_data)
     }
 
-    if (is.null(N)) N <- nrow(explainer$data)
-
     shap_values <- sapply(
         X = as.character(seq_len(nrow(new_observation))),
         FUN = function(i) {
@@ -327,7 +326,8 @@ use_treeshap <- function(explainer, new_observation, ...){
                         stopifnot(identical(dim(new_obs_mat), as.integer(c(1L, ncol(new_observation)))))
                         treeshap::treeshap(
                             unified_model = m,
-                            x = new_obs_mat
+                            x = new_obs_mat,
+                            ...
                         )$shaps
                     }
                 )