Skip to content

Instantly share code, notes, and snippets.

@erykml
Last active June 3, 2019 16:44
Show Gist options
  • Select an option

  • Save erykml/ae433fd708ca414519fa63d10cd9c07c to your computer and use it in GitHub Desktop.

Select an option

Save erykml/ae433fd708ca414519fa63d10cd9c07c to your computer and use it in GitHub Desktop.
from statsmodels.stats.outliers_influence import variance_inflation_factor
vif = [variance_inflation_factor(X_constant.values, i) for i in range(X_constant.shape[1])]
pd.DataFrame({'vif': vif[1:]}, index=X.columns).T
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment