bartDiag

Displays a selection of diagnostic plots for a BART model.

bartDiag(
  model,
  data,
  response,
  burnIn = 0,
  threshold = "Youden",
  pNorm = FALSE,
  showInterval = TRUE,
  combineFactors = FALSE
)

Arguments

model

a model created from either the BART, modelarts, or bartMachine package.
data

A dataframe used to build the model.
response

The name of the response for the fit.
burnIn

Trace plot will only show iterations above selected burn in value.
threshold

A dashed line on some plots to indicate a chosen threshold value (classification only). by default the Youden index is shown.
pNorm

apply pnorm to the y-hat data (classification only).
showInterval

LOGICAL if TRUE then show 5% and 95% quantile intervals on ROC an PC curves (classification only).
combineFactors

Whether or not to combine dummy variables (if present) in display.

Value

A selection of diagnostic plots

Examples

if (FALSE) {
bartDiag(model = my_model, response = data$response, burnIn = 100, data = my_data)
}