fixes for number of observations in model survshap

Author: krzyzinskim

R/model_survshap.R

@@ -59,6 +59,7 @@ model_survshap <- function(explainer, ...) {
 model_survshap.surv_explainer <- function(explainer,
                                           new_observation = NULL,
                                           y_true = NULL,
+                                          N = NULL,
                                           calculation_method = "kernelshap",
                                           aggregation_method = "integral",
                                           output_type = "survival",
@@ -98,6 +99,7 @@ model_survshap.surv_explainer <- function(explainer,
         explainer = explainer,
         new_observation = observations,
         output_type = output_type,
+        N = N,
         y_true = y_true,
         calculation_method = calculation_method,
         aggregation_method = aggregation_method

R/surv_shap.R

@@ -102,8 +102,8 @@ surv_shap <- function(explainer,
     # to display final object correctly, when is.matrix(new_observation) == TRUE
     res$variable_values <- as.data.frame(new_observation)
     res$result <- switch(calculation_method,
-                         "exact_kernel" = use_exact_shap(explainer, new_observation, output_type, ...),
-                         "kernelshap" = use_kernelshap(explainer, new_observation, output_type, ...),
+                         "exact_kernel" = use_exact_shap(explainer, new_observation, output_type, N, ...),
+                         "kernelshap" = use_kernelshap(explainer, new_observation, output_type, N, ...),
                          "treeshap" = use_treeshap(explainer, new_observation, ...),
                          stop("Only `exact_kernel`, `kernelshap` and `treeshap` calculation methods are implemented"))
     # quality-check here