man-roxygen/examples-IntegratedGradient.R

In bips-hb/innsight: Get the Insights of Your Neural Network

#' @examplesIf torch::torch_is_installed()
#' #----------------------- Example 1: Torch ----------------------------------
#' library(torch)
#'
#' # Create nn_sequential model and data
#' model <- nn_sequential(
#'   nn_linear(5, 12),
#'   nn_relu(),
#'   nn_linear(12, 2),
#'   nn_softmax(dim = 2)
#' )
#' data <- torch_randn(25, 5)
#' ref <- torch_randn(1, 5)
#'
#' # Create Converter
#' converter <- convert(model, input_dim = c(5))
#'
#' # Apply method IntegratedGradient
#' int_grad <- IntegratedGradient$new(converter, data, x_ref = ref)
#'
#' # You can also use the helper function `run_intgrad` for initializing
#' # an R6 IntegratedGradient object
#' int_grad <- run_intgrad(converter, data, x_ref = ref)
#'
#' # Print the result as a torch tensor for first two data points
#' get_result(int_grad, "torch.tensor")[1:2]
#'
#' # Plot the result for both classes
#' plot(int_grad, output_idx = 1:2)
#'
#' # Plot the boxplot of all datapoints and for both classes
#' boxplot(int_grad, output_idx = 1:2)
#'
#' # ------------------------- Example 2: Neuralnet ---------------------------
#' if (require("neuralnet")) {
#'   library(neuralnet)
#'   data(iris)
#'
#'   # Train a neural network
#'   nn <- neuralnet((Species == "setosa") ~ Petal.Length + Petal.Width,
#'     iris,
#'     linear.output = FALSE,
#'     hidden = c(3, 2), act.fct = "tanh", rep = 1
#'   )
#'
#'   # Convert the model
#'   converter <- convert(nn)
#'
#'   # Apply IntegratedGradient with a reference input of the feature means
#'   x_ref <- matrix(colMeans(iris[, c(3, 4)]), nrow = 1)
#'   int_grad <- run_intgrad(converter, iris[, c(3, 4)], x_ref = x_ref)
#'
#'   # Get the result as a dataframe and show first 5 rows
#'   get_result(int_grad, type = "data.frame")[1:5, ]
#'
#'   # Plot the result for the first datapoint in the data
#'   plot(int_grad, data_idx = 1)
#'
#'   # Plot the result as boxplots
#'   boxplot(int_grad)
#' }
#'
#' @examplesIf torch::torch_is_installed() & Sys.getenv("INNSIGHT_EXAMPLE_KERAS", unset = 0) == 1
#' # ------------------------- Example 3: Keras -------------------------------
#' if (require("keras") & keras::is_keras_available()) {
#'   library(keras)
#'
#'   # Make sure keras is installed properly
#'   is_keras_available()
#'
#'   data <- array(rnorm(10 * 32 * 32 * 3), dim = c(10, 32, 32, 3))
#'
#'   model <- keras_model_sequential()
#'   model %>%
#'     layer_conv_2d(
#'       input_shape = c(32, 32, 3), kernel_size = 8, filters = 8,
#'       activation = "softplus", padding = "valid") %>%
#'     layer_conv_2d(
#'       kernel_size = 8, filters = 4, activation = "tanh",
#'       padding = "same") %>%
#'     layer_conv_2d(
#'       kernel_size = 4, filters = 2, activation = "relu",
#'       padding = "valid") %>%
#'     layer_flatten() %>%
#'     layer_dense(units = 64, activation = "relu") %>%
#'     layer_dense(units = 2, activation = "softmax")
#'
#'   # Convert the model
#'   converter <- convert(model)
#'
#'   # Apply the IntegratedGradient method with a zero baseline and n = 20
#'   # iteration steps
#'   int_grad <- run_intgrad(converter, data,
#'     channels_first = FALSE,
#'     n = 20
#'   )
#'
#'   # Plot the result for the first image and both classes
#'   plot(int_grad, output_idx = 1:2)
#'
#'   # Plot the pixel-wise median of the results
#'   plot_global(int_grad, output_idx = 1)
#' }
#' @examplesIf torch::torch_is_installed() & Sys.getenv("RENDER_PLOTLY", unset = 0) == 1
#' #------------------------- Plotly plots ------------------------------------
#' if (require("plotly")) {
#'   # You can also create an interactive plot with plotly.
#'   # This is a suggested package, so make sure that it is installed
#'   library(plotly)
#'   boxplot(int_grad, as_plotly = TRUE)
#' }