| robust_vifs {randomForestVIP} | R Documentation |
Non-linear Variance Inflation Factors
Description
A list of data.frames and useful plots for user evaluations of the randomForest hyperparameter mtry.
Usage
robust_vifs(formula, data, model = randomForest, log10 = TRUE, num_var, ...)
Arguments
formula |
an object of class "formula" (or one that can be coerced to that class): a symbolic description of the model to be fitted. |
data |
an optional data frame containing the variables in the model. By default the variables are taken from the environment which the model is called from. |
model |
Model to use for extraction partial correlations. Possible model choices are rpart. |
log10 |
Applies a log10 transformation to VIFs when True. Default is True. |
num_var |
Optional integer argument for reducing the number of variables to the top 'num_var'. Should be an integer between 1 and the total number of predictor variables in the model or it should be a positive proportion of variables desired. |
... |
Additional arguments to be passed to models as needed. |
Value
A list of data.frames and useful plots for user evaluations of VIFs.
Examples
rv <- robust_vifs(Petal.Length ~ ., data = iris[-5], model = lm)
rv