margin | R Documentation |
Compute or plot the margin of predictions from a randomForest classifier.
margin(rf, observed)
## S3 method for class 'margin'
plot(x, sort=TRUE, ...)
rf |
an object of class |
observed |
the response variable (a factor) used to train
|
x |
an object of class |
sort |
Should the data be sorted by their class labels? |
... |
other graphical parameters to be passed to |
For margin
, the margin of observations from the
randomForest
classifier. 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.
Robert Gentlemen, with slight modifications by Andy Liaw
randomForest
set.seed(1)
data(iris)
iris.rf <- randomForest(Species ~ ., iris, keep.forest=FALSE)
plot(margin(iris.rf, iris$Species))
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.