diff --git a/DESCRIPTION b/DESCRIPTION index f240566..1cd6f48 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -1,15 +1,15 @@ Package: treeshap Title: Compute SHAP Values for Your Tree-Based Models Using the 'TreeSHAP' Algorithm -Version: 0.2.5.9000 +Version: 0.2.5.9001 Authors@R: c( - person("Konrad", "Komisarczyk", , "komisarczykkonrad@gmail.com", role = "aut"), - person("Pawel", "Kozminski", , "pkozminski99@gmail.com", role = "aut"), + person("Konrad", "Komisarczyk", email = "komisarczykkonrad@gmail.com", role = "aut"), + person("Pawel", "Kozminski", email = "pkozminski99@gmail.com", role = "aut"), person("Szymon", "Maksymiuk", role = "aut"), - person("Lorenz A.", "Kapsner", role = "ctb"), + person("Lorenz A.", "Kapsner", role = "ctb", comment = c(ORCID = "0000-0003-1866-860X")), person("Mikolaj", "Spytek", role = "ctb"), - person("Mateusz", "Krzyzinski", , "krzyzinskimateusz23@gmail.com", role = c("ctb", "cre")), - person("Przemyslaw", "Biecek", , "przemyslaw.biecek@gmail.com", role = c("aut", "cph")) + person("Mateusz", "Krzyzinski", email = "krzyzinskimateusz23@gmail.com", role = c("ctb", "cre")), + person("Przemyslaw", "Biecek", email = "przemyslaw.biecek@gmail.com", role = c("aut", "cph")) ) Description: An efficient implementation of the 'TreeSHAP' algorithm introduced by Lundberg et al., (2020) . diff --git a/R/unify_ranger_surv.R b/R/unify_ranger_surv.R index 604ff33..87ceb5c 100644 --- a/R/unify_ranger_surv.R +++ b/R/unify_ranger_surv.R @@ -61,7 +61,6 @@ #' unified_model_risk <- ranger_surv.unify(rf, train_x, type = "risk") #' shaps <- treeshap(unified_model_risk, train_x[1:2,]) #' -#' #' # compute shaps for 3 selected time points #' unified_model_surv <- ranger_surv.unify(rf, train_x, type = "survival", times = c(23, 50, 73)) #' for (i in 1:3) { diff --git a/man/ranger_surv.unify.Rd b/man/ranger_surv.unify.Rd index 34dabb1..e6bf9ca 100644 --- a/man/ranger_surv.unify.Rd +++ b/man/ranger_surv.unify.Rd @@ -62,7 +62,6 @@ rf <- ranger::ranger( unified_model_risk <- ranger_surv.unify(rf, train_x, type = "risk") shaps <- treeshap(unified_model_risk, train_x[1:2,]) - # compute shaps for 3 selected time points unified_model_surv <- ranger_surv.unify(rf, train_x, type = "survival", times = c(23, 50, 73)) for (i in 1:3) { diff --git a/src/RcppExports.o b/src/RcppExports.o index 0832527..65c9461 100644 Binary files a/src/RcppExports.o and b/src/RcppExports.o differ diff --git a/src/treeshap.o b/src/treeshap.o index 2e9e331..4cc161e 100644 Binary files a/src/treeshap.o and b/src/treeshap.o differ diff --git a/tests/testthat/test_ranger_surv.R b/tests/testthat/test_ranger_surv.R index 47802e5..18074d2 100644 --- a/tests/testthat/test_ranger_surv.R +++ b/tests/testthat/test_ranger_surv.R @@ -165,7 +165,6 @@ test_that('ranger_surv.unify (type = "survival"): mean prediction calculated usi #expect_true(all(abs((intercept_predict - intercept_covers) / intercept_predict) < 10**(-14))) expect_equal(intercept_predict, intercept_covers) - }) test_that('ranger_surv.unify (type = "survival"): covers correctness', {