ggplot.xgbm: Partial dependence plots for xgbm

Description Usage Arguments

View source: R/xgbm_methods.R

Description

Partial dependence plots for xgbm

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
## S3 method for class 'xgbm'
ggplot(
  data,
  mapping = NULL,
  i.var = 1,
  return.grid = FALSE,
  type = "auto",
  color = "blue",
  ...
)

Arguments

data

An object of class 'xgbm'.

mapping

Not used.

i.var

A number giving the predictor variable to plot (the function assumes there is no intercept), or a string giving the variable name.

return.grid

Whether to return the grid of data points and predicted values. Defaults to return.grid = FALSE and the plot is returned.

type

String with value 'auto' (the default), 'regression' or 'classification'. Passed through to pdp::partial.


harrysouthworth/mhdm documentation built on Feb. 4, 2022, 12:25 a.m.