plotVI: creates barplots for variable importances

In markusloecher/rfVarImpOOB: Unbiased Variable Importance for Random Forests

Description

creates barplots for variable importances

Usage

plotVI(VIbench, order_by = "Gini_OOB", decreasing = TRUE)

Arguments

VIbench	matrix with importance scores as returned by GiniImportanceForest
order_by	how to order
decreasing	which direction to sort

Author(s)

Markus Loecher <[e-mail]>

Examples

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)

markusloecher/rfVarImpOOB documentation built on July 5, 2020, 6:50 p.m.