R/print_ceteris_paribus.R

In pbiecek/ceterisParibus2: Ceteris Paribus Profiles

#' Prints Individual Variable Explainer Summary
#'
#' @param x an individual variable profile explainer produced with the `individual_variable_profile()` function
#' @param ... other arguments that will be passed to `head()`
#'
#' @export
#'
#' @examples
#' library("DALEX2")
#'  \dontrun{
#' library("randomForest")
#' set.seed(59)
#'
#' apartments_rf_model <- randomForest(m2.price ~ construction.year + surface + floor +
#'       no.rooms + district, data = apartments)
#'
#' explainer_rf <- explain(apartments_rf_model,
#'       data = apartments_test[,2:6], y = apartments_test$m2.price)
#'
#' apartments_small <- select_sample(apartments_test, 10)
#'
#' cp_rf <- individual_variable_profile(explainer_rf, apartments_small)
#' cp_rf
#' }
print.individual_variable_profile_explainer <- function(x, ...) {
  cat("Top profiles    : \n")
  class(x) <- "data.frame"
  print(head(x, ...))

  cat("\n\nTop observations:\n")
  print(head(as.data.frame(attr(x, "observations")), ...))
  return(invisible(NULL))
}