getVI {Rforestry} | R Documentation |
getVI-forestry
Description
Calculate the percentage increase in OOB error of the forest when each feature is shuffled.
Usage
getVI(object, noWarning, metric = "mse", seed = 1)
Arguments
object |
A 'forestry' object. |
noWarning |
flag to not display warnings |
metric |
A parameter to determine how the predictions of the forest with a permuted variable are compared to the predictions of the standard forest. Must be one of c("mse","auc","tnr"), "mse" gives the percentage increase in mse when the feature is permuted, "auc" gives the percentage decrease in AUC when the feature is permuted, and "tnr" gives the percentage decrease in TNR when the TPR is 99% when the feature is permuted. |
seed |
A parameter to seed the random number generator for shuffling the features of X. |
Value
The variable importance of the forest.
Note
Pass a seed to this function so it is possible to replicate the vector permutations used when measuring feature importance.