RocPlot {LOGANTree}

ROC Curves Plot

Description

ROC Curves Plot

Usage

RocPlot(ModelObject, testdata, outcome, reflevel)

Arguments

ModelObject

An object obtained from TreeModels() or TreeModelsAllSteps() functions.

testdata

A testing dataset.

outcome

A character string with the name of the binary outcome variable.

reflevel

A character string with the quoted reference level of outcome.

Value

This function returns a plot with ROC curves for the selected tree-based models (i.e., decision tree, random forest, or gradient boosting).

Examples

colnames(training)[14] <- "perf"
colnames(testing)[14] <- "perf"
ensemblist <- TreeModels(traindata = training,
methodlist = c("dt", "gbm","rf"),checkprogress = TRUE)

RocPlot(ModelObject = ensemblist$ModelObject, testdata = testing,
outcome = "perf", reflevel = "incorrect")