plotVI2: creates barplots for variable importances

In markusloecher/rfVarImpOOB: Unbiased Variable Importance for Random Forests

Description

creates barplots for variable importances including permutation scores

Usage

plotVI2(VIbench, col2plot = paste0("IG_pgOOB", 2), 


    Gini_inbag_col = "IG_pgOOB4", decreasing = TRUE, 


    with_MDA = TRUE, ordered_by = "inbag", horizontal = TRUE, 


    fill = "order", labelSize = 10, nrow = 3)

Arguments

VIbench	matrix with importance scores as returned by GiniImportanceForest
col2plot
Gini_inbag_col
decreasing	which direction to sort
with_MDA	also visualize mean decrease in accuracy (permutation importance)
ordered_by	how to order
horizontal	horizontal barplot instead of vertical ?
fill	fill style for barplots; use e.g. shQuote("blue") to pass color strings
labelSize	size of axis labels
nrow	number of rows of ploztz arrangement

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")


plotVI2(VI_Titanic,decreasing = TRUE)

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