pred_plot {radiant.model}

Prediction Plots

Description

Prediction Plots

Usage

pred_plot(
  x,
  plot_list = list(),
  incl,
  incl_int,
  fix = TRUE,
  hline = TRUE,
  nr = 20,
  minq = 0.025,
  maxq = 0.975
)

Arguments

x

Return value from a model

plot_list

List used to store plots

incl

Which variables to include in prediction plots

incl_int

Which interactions to investigate in prediction plots

fix

Set the desired limited on yhat or have it calculated automatically. Set to FALSE to have y-axis limits set by ggplot2 for each plot

hline

Add a horizontal line at the average of the target variable. When set to FALSE no line is added. When set to a specific number, the horizontal line will be added at that value

nr

Number of values to use to generate predictions for a numeric explanatory variable

minq

Quantile to use for the minimum value for simulation of numeric variables

maxq

Quantile to use for the maximum value for simulation of numeric variables

Details

Faster, but less robust, alternative for PDP plots. Variable values not included in the prediction are set to either the mean or the most common value (level)