From 3beacd7b15abe64232fa1a8f388654ae61816eb5 Mon Sep 17 00:00:00 2001 From: NienkeVanTKlooster Date: Tue, 16 Jul 2019 15:36:04 +0200 Subject: [PATCH] Update a_dataframe.R --- R/a_dataframe.R | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/R/a_dataframe.R b/R/a_dataframe.R index ab1ffc0..ba269d7 100644 --- a/R/a_dataframe.R +++ b/R/a_dataframe.R @@ -100,7 +100,7 @@ lime.data.frame <- function(x, model, preprocess = NULL, bin_continuous = TRUE, explain.data.frame <- function(x, explainer, labels = NULL, n_labels = NULL, n_features, n_permutations = 5000, feature_select = 'auto', dist_fun = 'gower', - kernel_width = NULL, gower_pow = 1, ...) { + kernel_width = NULL, gower_pow = 1, feature_imp = NULL, ...) { assert_that(is.data_frame_explainer(explainer)) m_type <- model_type(explainer) o_type <- output_type(explainer) @@ -129,7 +129,7 @@ explain.data.frame <- function(x, explainer, labels = NULL, n_labels = NULL, res <- lapply(seq_along(case_ind), function(ind) { i <- case_ind[[ind]] if (dist_fun == 'gower') { - sim <- 1 - (gower_dist(case_perm[i[1], , drop = FALSE], case_perm[i, , drop = FALSE])) ^ gower_pow + sim <- 1 - (gower_dist(case_perm[i[1], , drop = FALSE], case_perm[i, , drop = FALSE], weights = feature_imp)) ^ gower_pow } perms <- numerify(case_perm[i, ], explainer$feature_type, explainer$bin_continuous, explainer$bin_cuts) if (dist_fun != 'gower') {