plotVI | R Documentation |
creates barplots for variable importances
plotVI(VIbench, order_by = "Gini_OOB", decreasing = TRUE)
VIbench |
matrix with importance scores as returned by GiniImportanceForest |
order_by |
how to order |
decreasing |
which direction to sort |
Markus Loecher <Markus.Loecher@gmail.com>
data("titanic_train", package = "rfVarImpOOB", envir = environment()) set.seed(123) ranRows=sample(nrow(titanic_train), 300) data=titanic_train[ranRows,] RF = randomForest::randomForest(formula = Survived ~ Sex + Pclass + PassengerId, data=data, ntree=5,importance=TRUE, mtry=3,keep.inbag=TRUE, nodesize = 20) data$Survived = as.numeric(data$Survived)-1 VI_Titanic = GiniImportanceForest(RF, data,ylab="Survived") plotVI(VI_Titanic,decreasing = TRUE)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.