radiant.model (version 1.6.12)

plot.gbt: Plot method for the gbt function

Description

Plot method for the gbt function

Usage

# S3 method for gbt
plot(
  x,
  plots = "",
  nrobs = Inf,
  incl = NULL,
  incl_int = NULL,
  hline = TRUE,
  pdp_range = c(0.025, 0.975),
  minq = NULL,
  maxq = NULL,
  shiny = FALSE,
  custom = FALSE,
  ...
)

Arguments

x

Return value from gbt

plots

Plots to produce for the specified Gradient Boosted Tree model. Use "" to avoid showing any plots (default). Options are ...

nrobs

Number of data points to show in scatter plots (-1 for all)

incl

Which variables to include in a coefficient plot or PDP plot

incl_int

Which interactions to investigate in PDP plots

hline

Add a dashed horizontal line at the mean response (TRUE/FALSE or a numeric value)

pdp_range

Numeric vector c(lo, hi) giving the percentile range for PDP/Prediction plot x-axes (default c(0.025, 0.975))

minq

Deprecated. Use pdp_range[1] instead

maxq

Deprecated. Use pdp_range[2] instead

shiny

Did the function call originate inside a shiny app

custom

Logical (TRUE, FALSE) to indicate if ggplot object (or list of ggplot objects) should be returned. This option can be used to customize plots (e.g., add a title, change x and y labels, etc.). See examples and https://ggplot2.tidyverse.org for options.

...

further arguments passed to or from other methods

Details

See https://radiant-rstats.github.io/docs/model/gbt.html for an example in Radiant

See Also

gbt to generate results

summary.gbt to summarize results

predict.gbt for prediction

Examples

result <- gbt(
  titanic, "survived", c("pclass", "sex"),
  early_stopping_rounds = 0, nthread = 1
)
plot(result)