Description Usage Arguments Value Author(s) Examples
The function takes a dataset with the starting variables and target only. The vif is calculated and if the maximum vif value is more than the threshold, the variable is dropped from the model and the vif's are recomputed. These steps of computing vif and dropping variable keep iterating till the maximum vif value is less than or equal to the threshold.
1 | vif_filter(base, target, threshold = 2)
|
base |
input dataframe with set of final variables only along with target |
target |
column / field name for the target variable to be passed as string (must be 0/1 type) |
threshold |
threshold value for vif (default value is 2) |
An object of class "vif_filter" is a list containing the following components:
vif_table |
vif table post vif filtering |
model |
the model used for vif calculation |
retain_var_list |
variables remaining in the model post vif filter as an array |
dropped_var_list |
variables dropped from the model in vif filter step |
threshold |
threshold |
Arya Poddar <aryapoddar290990@gmail.com>
1 2 3 4 5 6 7 8 9 10 | data <- iris
suppressWarnings(RNGversion('3.5.0'))
set.seed(11)
data$Y <- sample(0:1,size=nrow(data),replace=TRUE)
vif_data_list <- vif_filter(base = data,target = "Y")
vif_data_list$vif_table
vif_data_list$model
vif_data_list$retain_var_list
vif_data_list$dropped_var_list
vif_data_list$threshold
|
vif.log_model.
Sepal.Length 3.332934
Sepal.Width 1.285032
Petal.Width 3.760406
Call: glm(formula = as.formula(formula), family = "binomial", data = base2)
Coefficients:
(Intercept) Sepal.Length Sepal.Width Petal.Width
-0.2531 -0.2818 0.2126 0.7191
Degrees of Freedom: 149 Total (i.e. Null); 146 Residual
Null Deviance: 202.7
Residual Deviance: 198.4 AIC: 206.4
[1] "Sepal.Length" "Sepal.Width" "Petal.Width" "Y"
[1] "Petal.Length" "Species"
[1] 2
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.