/
print_explainer.R
33 lines (32 loc) · 1.03 KB
/
print_explainer.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#' Print Explainer Summary
#'
#' @param x a model explainer created with the `explain` function
#' @param ... other parameters
#'
#' @export
#' @import ggplot2
#'
#' @examples
#'
#' aps_lm_model4 <- lm(m2.price~., data = apartments)
#' aps_lm_explainer4 <- explain(aps_lm_model4, data = apartments, y = apartments$m2.price,
#' label = "model_4v")
#' aps_lm_explainer4
#'
#' \donttest{
#' library("ranger")
#' titanic_ranger_model <- ranger(survived~., data = titanic_imputed, num.trees = 50,
#' probability = TRUE)
#' explainer_ranger <- explain(titanic_ranger_model, data = titanic_imputed[,-8],
#' y = titanic_imputed$survived,
#' label = "model_ranger")
#' explainer_ranger
#' }
#'
print.explainer <- function(x, ...) {
cat("Model label: ", x$label, "\n")
cat("Model class: ", paste(x$class, collapse = ","), "\n")
cat("Data head :\n")
print(head(x$data,2))
return(invisible(NULL))
}