Fix issue #72

ModelOriented · Mar 24, 2023 · 42b7a97 · 42b7a97
1 parent 76993e0
commit 42b7a97
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/plot_surv_lime.R b/R/plot_surv_lime.R
@@ -39,7 +39,7 @@ plot.surv_lime <- function(x,
     if (!type %in% c("coefficients", "local_importance"))
         stop("Type should be one of `coefficients`, `local_importance`")
 
-    local_importance <- as.numeric(x$result * x$variable_values)
+    local_importance <- as.numeric(x$result) * as.numeric(x$variable_values)
     df <- data.frame(variable_names = names(x$variable_values),
                      variable_values = as.numeric(x$variable_values),
                      beta = as.numeric(x$result),

diff --git a/R/print.R b/R/print.R
@@ -33,7 +33,7 @@ print.surv_feature_importance <- function(x, ...) {
 print.surv_lime <- function(x, ...) {
     res <- x$result
 
-    print_result <- rbind(beta = res, `X` = x$variable_values, `local importance (X*beta)` = res * x$variable_values)
+    print_result <- rbind(beta = res, `X` = x$variable_values, `local importance (X*beta)` = res * as.numeric(x$variable_values))
     cat("SurvLIME for observation:\n\n")
     print.data.frame(x$variable_values, row.names = FALSE)
     cat("\n")