randomForest (version 4.6-6)

margin: Margins of randomForest Classifier

Description

Compute or plot the margin of predictions from a randomForest classifier.

Usage

## S3 method for class 'randomForest':
margin(x, ...)
## S3 method for class 'default':
margin(x, observed, ...)
## S3 method for class 'margin':
plot(x, sort=TRUE, ...)

Arguments

x
an object of class randomForest, whose type is not regression, or a matrix of predicted probabilities, one column per class and one row per observation. For the pl
observed
the true response corresponding to the data in x.
sort
Should the data be sorted by their class labels?
...
other graphical parameters to be passed to plot.default.

Value

  • For margin, the margin of observations from the randomForest classifier (or whatever classifier that produced the predicted probability matrix given to margin). The margin of a data point is defined as the proportion of votes for the correct class minus maximum proportion of votes for the other classes. Thus under majority votes, positive margin means correct classification, and vice versa.

See Also

randomForest

Examples

Run this code
set.seed(1)
data(iris)
iris.rf <- randomForest(Species ~ ., iris, keep.forest=FALSE)
plot(margin(iris.rf))

Run the code above in your browser using DataCamp Workspace