Skip to content

Commit 0524683

Browse files
committed
use rep.int() in one of the bottlenecks
1 parent 1a74936 commit 0524683

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

R/utils_kernelshap.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ kernelshap_one <- function(x, v1, object, pred_fun, feature_names, bg_w, exact,
1515

1616
# Most expensive part
1717
vz <- get_vz( # (m_ex x K)
18-
X = x[rep(1L, times = nrow(bg_X_exact)), , drop = FALSE], # (m_ex*n_bg x p)
18+
X = x[rep.int(1L, times = nrow(bg_X_exact)), , drop = FALSE],# (m_ex*n_bg x p)
1919
bg = bg_X_exact, # (m_ex*n_bg x p)
2020
Z = Z, # (m_ex x p)
2121
object = object,

R/utils_permshap.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ shapley_weights <- function(p, ell) {
2828
permshap_one <- function(x, v1, object, pred_fun, bg_w, v0, precalc, ...) {
2929
Z <- precalc[["Z"]] # ((m_ex+2) x K)
3030
vz <- get_vz( # (m_ex x K)
31-
X = x[rep(1L, times = nrow(precalc[["bg_X_rep"]])), , drop = FALSE], # (m_ex*n_bg x p)
31+
X = x[rep.int(1L, times = nrow(precalc[["bg_X_rep"]])), , drop = FALSE], # (m_ex*n_bg x p)
3232
bg = precalc[["bg_X_rep"]], # (m_ex*n_bg x p)
3333
Z = Z[2:(nrow(Z) - 1L), , drop = FALSE], # (m_ex x p)
3434
object = object,

0 commit comments

Comments
 (0)